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