Searched defs:indices0 (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | dynamic_stitch_op.cc | 61 static bool SameExtraShape(const Tensor& data0, const Tensor& indices0, in SameExtraShape()
103 const Tensor& indices0 = (*indices_inputs)[0]; in CheckArgsAndAllocateResult() local
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_index_test.cpp | 352 optional<Tensor> indices0[] = { in TEST_F() local
411 optional<Tensor> indices0[] = {optional<Tensor>()}; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | dynamic_stitch_op.cc | 206 const TensorShape& indices0, in SameExtraShape()
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_generated_ops.cc | 4419 auto indices0 = at::argsort(self0, 1, true); in TEST() local
6138 auto indices0 = at::randint(0, 1, {2, 2, 2}, at::kLong); in TEST() local
6172 auto indices0 = at::randint(0, 1, {2, 2, 2, 2}, at::kLong); in TEST() local
|