Searched defs:indices1 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 208 const TensorShape& indices1) { in SameExtraShape()
|
| /external/executorch/kernels/test/ |
| D | op_index_test.cpp | 369 optional<Tensor> indices1[] = { in TEST_F() local
414 optional<Tensor> indices1[] = {optional<Tensor>(), optional<Tensor>()}; in TEST_F() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 167 Eigen::DSizes<ptrdiff_t, 2> indices1(1,2); in test_slice_in_expr() local
490 Eigen::DSizes<ptrdiff_t, 3> indices1(1,2,3); in test_empty_slice() local
|
| /external/angle/src/tests/gl_tests/ |
| D | DrawElementsTest.cpp | 681 const GLubyte indices1[] = {0, 0, 0, 0, 0, 0}; in TEST_P() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 62 const Tensor& data1, const Tensor& indices1) { in SameExtraShape()
|
| /external/pytorch/benchmarks/static_runtime/ |
| D | test_generated_ops.cc | 4431 auto indices1 = at::argsort(self1, 1, true); in TEST() local
6150 auto indices1 = at::randint(0, 1, {3, 3, 3}, at::kLong); in TEST() local
6184 auto indices1 = at::randint(0, 1, {3, 3, 3, 3}, at::kLong); in TEST() local
|
| /external/pytorch/test/quantization/jit/ |
| D | test_quantize_jit.py | 3426 def forward(self, indices1, offsets1, indices2, offsets2): argument
|