コード例 #1
0
def test_map_increase_rank_2d():
  expect_shape(simple_map, [mat], Shape([Var(0), Var(1)]))
コード例 #2
0
def test_simple_map_2d():
  expect_shape(simple_map, [mat], Shape([Var(0), Var(1)]))
コード例 #3
0
def test_map_increase_rank_1d():
  expect_shape(map_increase_rank, [vec], Shape([Var(0), 2]))
コード例 #4
0
def test_increase_rank_3d():
  expect_shape(increase_rank, [mat], Shape([2, Var(0), Var(1)]))
コード例 #5
0
def test_simple_map_1d():
  expect_shape(simple_map, [vec], Shape([Var(0)]))
コード例 #6
0
def test_increase_rank_2d():
  expect_shape(increase_rank, [vec], Shape([2, Var(0)]))
コード例 #7
0
def test_increase_rank_1d():
  expect_shape(increase_rank, [1], Shape([2]))
コード例 #8
0
def test_ident_2d():
  expect_shape(ident, [mat], Shape([Var(0), Var(1)]))
コード例 #9
0
def test_ident_1d():
  expect_shape(ident, [vec], Shape([Var(0)]))
コード例 #10
0
def test_array_literal():
  expect_shape(array_literal, [], Shape([3]))