Lines Matching defs:indices
29 const Tensor& indices, in op_embedding_out()
56 Tensor indices = tfl.make( in test_dtype() local
97 Tensor indices = tf_indices.make({2, 4}, {1, 2, 4, 5, 4, 3, 2, 9}); in test_dynamic_shape() local
128 Tensor indices = tfl.make({1}, {1}); in TEST_F() local
170 Tensor indices = tfl.make({1, 2, 3}, {1, 0, 2, 3, 4, 0}); in TEST_F() local
207 Tensor indices = tfl.make({2, 2}, {1, 0, 2, 3}); in TEST_F() local
241 Tensor indices = tfl.make({2, 2}, {1, 0, 2, 3}); in TEST_F() local
272 Tensor indices = tfl.make({2, 2}, {1, 0, 2, 3}); in TEST_F() local
336 Tensor indices = tfl.make({2, 2}, {2, 0, 2, 4}); in TEST_F() local
366 Tensor indices = tfl.make({}, {3}); in TEST_F() local
403 Tensor indices = tfl.make({3, 0}, {}); in TEST_F() local