Lines Matching refs:Tensor
24 using exec_aten::Tensor;
27 using OptTensorArrayRef = ArrayRef<optional<Tensor>>;
31 Tensor& op_index_tensor_out( in op_index_tensor_out()
32 const Tensor& input, in op_index_tensor_out()
34 Tensor& out) { in op_index_tensor_out()
36 c10::List<std::optional<at::Tensor>> indices_list(indices); in op_index_tensor_out()
55 Tensor x = tf.make( in test_dtype()
75 optional<Tensor> indices[] = { in test_dtype()
76 optional<Tensor>(tfl.make({2}, {0, 1})), in test_dtype()
77 optional<Tensor>(tfl.make({2}, {1, 0})), in test_dtype()
78 optional<Tensor>(tfl.make({2}, {2, 3}))}; in test_dtype()
80 optional<Tensor> indices_mixed[] = { in test_dtype()
81 optional<Tensor>(tfl.make({2}, {0, 1})), in test_dtype()
82 optional<Tensor>(tfb.make({2}, {false, true})), in test_dtype()
83 optional<Tensor>(tfl.make({2}, {2, 3}))}; in test_dtype()
87 Tensor out_0 = tfo.zeros(out_size); in test_dtype()
88 Tensor ret_0 = op_index_tensor_out(x, /*indices=*/indices, out_0); in test_dtype()
95 Tensor out_0_with_mixed = tfo.zeros(out_size); in test_dtype()
96 Tensor ret_0_with_mixed = in test_dtype()
117 const Tensor& x, in run_test_cases()
119 const Tensor& expected) { in run_test_cases()
125 Tensor out = tf.ones(out_size); in run_test_cases()
127 Tensor ret = op_index_tensor_out(x, indices, out); in run_test_cases()
141 Tensor x = tf.make( in TEST_F()
157 Tensor indices = tfb.make( in TEST_F()
173 Tensor expected = tf.make( in TEST_F()
188 Tensor x = tf.make( in TEST_F()
206 optional<Tensor> indices[] = { in TEST_F()
207 optional<Tensor>(tfl.make({1}, {1})), in TEST_F()
208 optional<Tensor>(tfl.make({1}, {0})), in TEST_F()
209 optional<Tensor>(tfl.make({2}, {1, 2}))}; in TEST_F()
211 optional<Tensor> indices_int[] = { in TEST_F()
212 optional<Tensor>(tfi.make({1}, {1})), in TEST_F()
213 optional<Tensor>(tfi.make({1}, {0})), in TEST_F()
214 optional<Tensor>(tfi.make({2}, {1, 2}))}; in TEST_F()
216 optional<Tensor> indices_negative[] = { in TEST_F()
217 optional<Tensor>(tfl.make({1}, {-1})), in TEST_F()
218 optional<Tensor>(tfl.make({1}, {0})), in TEST_F()
219 optional<Tensor>(tfl.make({2}, {-3, -2}))}; in TEST_F()
221 optional<Tensor> indices_bool[] = { in TEST_F()
222 optional<Tensor>(tfb.make({2}, {false, true})), in TEST_F()
223 optional<Tensor>(tfb.make({3}, {true, false, false})), in TEST_F()
224 optional<Tensor>(tfl.make({2}, {-3, -2}))}; in TEST_F()
226 optional<Tensor> indices_mixed[] = { in TEST_F()
227 optional<Tensor>(tfb.make({2}, {false, true})), in TEST_F()
228 optional<Tensor>(tfl.make({1}, {0})), in TEST_F()
229 optional<Tensor>(tfl.make({2}, {-3, -2}))}; in TEST_F()
234 Tensor expected = tf.make( in TEST_F()
251 Tensor x = tf.make( in TEST_F()
267 optional<Tensor> indices[] = { in TEST_F()
268 optional<Tensor>(tfl.make({1, 2}, {0, 0})), in TEST_F()
269 optional<Tensor>(tfl.make({1, 2}, {1, 1})), in TEST_F()
270 optional<Tensor>(tfl.make({1, 2}, {2, 2}))}; in TEST_F()
275 Tensor expected = tf.make( in TEST_F()
290 Tensor x = tf.make( in TEST_F()
309 optional<Tensor> indices[] = { in TEST_F()
310 optional<Tensor>(tfl.make({1}, {1})), in TEST_F()
311 optional<Tensor>(tfl.make({2}, {0, 1}))}; in TEST_F()
313 optional<Tensor> indices_mixed[] = { in TEST_F()
314 optional<Tensor>(tfi.make({1}, {-1})), in TEST_F()
315 optional<Tensor>(tfb.make({3}, {true, true, false}))}; in TEST_F()
320 Tensor expected = tf.make( in TEST_F()
337 Tensor x = tf.make( in TEST_F()
352 optional<Tensor> indices0[] = { in TEST_F()
353 optional<Tensor>(), in TEST_F()
354 optional<Tensor>(tfl.make({1}, {1})), in TEST_F()
355 optional<Tensor>(tfl.make({2}, {0, 1}))}; in TEST_F()
358 Tensor expected0 = tf.make( in TEST_F()
369 optional<Tensor> indices1[] = { in TEST_F()
370 optional<Tensor>(tfl.make({1}, {1})), in TEST_F()
371 optional<Tensor>(), in TEST_F()
372 optional<Tensor>(tfl.make({2}, {0, 1}))}; in TEST_F()
375 Tensor expected1 = tf.make( in TEST_F()
386 optional<Tensor> indices2[] = { in TEST_F()
387 optional<Tensor>(tfl.make({1}, {1})), in TEST_F()
388 optional<Tensor>(tfl.make({2}, {0, 1})), in TEST_F()
389 optional<Tensor>()}; in TEST_F()
392 Tensor expected2 = tf.make( in TEST_F()
410 Tensor x = tf.make({2, 3}, {1., 2., 3., 4., 5., 6.}); in TEST_F()
411 optional<Tensor> indices0[] = {optional<Tensor>()}; in TEST_F()
414 optional<Tensor> indices1[] = {optional<Tensor>(), optional<Tensor>()}; in TEST_F()
417 optional<Tensor> indices2[] = { in TEST_F()
418 optional<Tensor>(), optional<Tensor>(), optional<Tensor>()}; in TEST_F()
419 Tensor out = tf.ones({2, 3}); in TEST_F()
431 Tensor x = tf.make({2}, {1., 2.}); in TEST_F()
433 Tensor out = tf.zeros({2}); in TEST_F()
462 Tensor x = tf.ones({1, 1, 1}); in TEST_F()
463 Tensor out = tf.zeros({1, 1, 1}); in TEST_F()
464 Tensor index = tfl.make({1}, {5}); in TEST_F()
474 Tensor x = tf.ones({1, 1, 1}); in TEST_F()
475 Tensor out = tf.zeros({1, 1, 1}); in TEST_F()
476 Tensor index = tfl.make({1}, {-5}); in TEST_F()
486 Tensor x = tf.ones({1, 1, 1}); in TEST_F()
487 Tensor out = tf.zeros({1, 1, 1}); in TEST_F()
488 Tensor index = tfb.make({3}, {true, false, false}); in TEST_F()
498 Tensor x = tf.ones({4}); in TEST_F()
499 Tensor out = tf.zeros({1}); in TEST_F()
500 Tensor index = tfb.make({1}, {true}); in TEST_F()
511 Tensor x = tf.ones({4, 4}); in TEST_F()
512 Tensor out = tf.zeros({9}); in TEST_F()
513 Tensor index = tfb.ones({3, 3}); in TEST_F()
524 Tensor x = tf.zeros({2, 4, 7, 5}); in TEST_F()
525 Tensor index = tfl.make({1}, {3}); in TEST_F()
528 Tensor out = tf.zeros({2, 4}); in TEST_F()
538 Tensor x = tf.zeros({2, 4, 7, 5}); in TEST_F()
539 Tensor index = tff.make({1}, {3}); in TEST_F()
541 Tensor out = tf.zeros({1, 4, 7, 5}); in TEST_F()
551 Tensor x = tf.zeros({2, 4, 7, 5}); in TEST_F()
553 optional<Tensor> indices[] = { in TEST_F()
554 optional<Tensor>(tfl.make({3}, {1, 1, 1,})), in TEST_F()
555 optional<Tensor>(tfl.make({2}, {1, 2}))}; in TEST_F()
557 Tensor out = tf.ones({3, 7, 5}); in TEST_F()
571 Tensor x = tf.zeros({4, 4}); in TEST_F()
573 optional<Tensor> indices[] = { in TEST_F()
574 optional<Tensor>(tfl.make({2, 2}, {1, 1, 1, 1,})), in TEST_F()
575 optional<Tensor>(tfl.make({1, 2}, {3, 0,}))}; in TEST_F()
577 Tensor out = tf.ones({4}); in TEST_F()
593 Tensor x = tf.make( in TEST_F()
610 optional<Tensor> indices[] = { in TEST_F()
611 optional<Tensor>(tfl.make({1, 2}, {0, 1})), in TEST_F()
612 optional<Tensor>(tfl.make({1, 2}, {2, 1})), in TEST_F()
613 optional<Tensor>(tfl.make({1, 2}, {2, 2}))}; in TEST_F()
615 Tensor out = in TEST_F()
618 Tensor expected = tf.make( in TEST_F()
626 Tensor ret = op_index_tensor_out(x, indices, out); in TEST_F()