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