示例#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]))