| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GatherFixture.h | 51 void setup(TensorShape input_shape, TensorShape indices_shape, int axis, DataType data_type) in setup()
65 …dices(U &&indices, const TensorShape &input_shape, uint32_t actual_axis, TensorShape indices_shape) in generate_indices()
81 const TensorShape indices_shape) in compute_target()
120 const TensorShape indices_shape) in compute_reference()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_reduce_sum_op_test.cc | 54 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
96 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | sparse_dense_binary_op_shared_test.cc | 100 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
129 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
156 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
188 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | scatter_nd_util.cc | 23 const TensorShape& indices_shape, in ValidateScatterNdUpdateShape()
|
| D | sparse_add_op_test.cc | 64 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | one_hot_op.cc | 56 const TensorShape& indices_shape = indices.shape(); in Compute() local
|
| D | ragged_gather_op_test.cc | 35 const TensorShape& indices_shape, const std::vector<INDEX_TYPE>& indices, in BuildRaggedGatherGraph()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | state_ops.cc | 118 ShapeHandle indices_shape = c->input(1); in ScatterUpdateShape() local
142 ShapeHandle indices_shape; in ScatterNdUpdateShape() local
|
| D | resource_variable_ops.cc | 266 ShapeHandle indices_shape = c->input(1); in __anon620315450402() local
327 ShapeHandle indices_shape = c->input(1); in ResourceScatterUpdateShape() local
|
| D | math_ops.cc | 1175 ShapeHandle indices_shape; in SparseSegmentReductionShapeFn() local
1199 ShapeHandle indices_shape; in SparseSegmentReductionGradShapeFn() local
1237 ShapeHandle indices_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 37 const xla::XlaOp& indices, const TensorShape& indices_shape, in XlaGather()
161 auto indices_shape = context->InputShape(1); in XlaGatherWithBatchDimsOpImpl() local
280 TensorShape indices_shape = context->InputShape(1); in Compile() local
|
| D | scatter_nd_op.cc | 35 const TensorShape& indices_shape, in ValidateUpdateShape()
92 TensorShape indices_shape = context->InputShape(0); in Compile() local
151 TensorShape indices_shape = context->InputShape(1); in CompileTensorScatter() local
|
| D | one_hot_op.cc | 33 const TensorShape indices_shape = ctx->InputShape(0); in Compile() local
|
| D | sparse_to_dense_op.cc | 31 const TensorShape indices_shape = context->InputShape(0); in Compile() local
|
| D | dynamic_stitch_op.cc | 71 TensorShape indices_shape; in Compile() local
|
| D | segment_reduction_ops.cc | 58 TensorShape indices_shape = ctx->InputShape(1); in Compile() local
|
| D | tensor_array_ops.cc | 306 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
374 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils_test.cc | 262 const Shape& indices_shape = args[1].shape(); in XLA_TEST_F() local
300 const Shape& indices_shape = args[1].shape(); in XLA_TEST_F() local
|
| /external/ComputeLibrary/tests/datasets/ |
| D | GatherDataset.h | 92 void add_config(TensorShape input_shape, TensorShape indices_shape, int axis) in add_config()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 48 TensorShape indices_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_one_hot_op_test.py | 151 indices_shape, argument
|
| D | ragged_gather_op_test.py | 394 indices_shape, argument
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_unpooling_2d.cc | 76 const RuntimeShape indices_shape = GetTensorShape(indices); in Prepare() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense_test.cc | 38 SparseToDenseOpModel(std::initializer_list<int> indices_shape, in SparseToDenseOpModel()
|
| D | embedding_lookup_sparse_test.cc | 40 std::initializer_list<int> indices_shape, in EmbeddingLookupSparseOpModel()
|