Searched refs:dim_rev (Results 1 – 1 of 1) sorted by relevance
24 array<bool, 4> dim_rev; in test_simple_reverse() local25 dim_rev[0] = false; in test_simple_reverse()26 dim_rev[1] = true; in test_simple_reverse()27 dim_rev[2] = true; in test_simple_reverse()28 dim_rev[3] = false; in test_simple_reverse()31 reversed_tensor = tensor.reverse(dim_rev); in test_simple_reverse()48 dim_rev[0] = true; in test_simple_reverse()49 dim_rev[1] = false; in test_simple_reverse()50 dim_rev[2] = false; in test_simple_reverse()51 dim_rev[3] = false; in test_simple_reverse()[all …]