Home
last modified time | relevance | path

Searched defs:indices_shape (Results 1 – 25 of 53) sorted by relevance

123

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGatherFixture.h51 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_dense_binary_op_shared_test.cc100 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
H A Dsparse_reduce_sum_op_test.cc54 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
96 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
H A Dexample_parsing_ops.cc1038 TensorShape indices_shape({num_elements, 1}); in Compute() local
1051 TensorShape indices_shape({0, 1}); in Compute() local
1161 TensorShape indices_shape({total_num_features, 2}); in Compute() local
H A Dscatter_nd_util.cc23 const TensorShape& indices_shape, in ValidateScatterNdUpdateShape()
H A Dsparse_add_op_test.cc64 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
H A Done_hot_op.cc56 const TensorShape& indices_shape = indices.shape(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc37 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
H A Dscatter_nd_op.cc35 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
H A Dtensor_array_ops.cc306 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
374 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
H A Done_hot_op.cc33 const TensorShape indices_shape = ctx->InputShape(0); in Compile() local
H A Dsparse_to_dense_op.cc31 const TensorShape indices_shape = context->InputShape(0); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dstate_ops.cc118 ShapeHandle indices_shape = c->input(1); in ScatterUpdateShape() local
142 ShapeHandle indices_shape; in ScatterNdUpdateShape() local
H A Dresource_variable_ops.cc266 ShapeHandle indices_shape = c->input(1); in __anon9faba5f60402() local
327 ShapeHandle indices_shape = c->input(1); in ResourceScatterUpdateShape() local
H A Dmath_ops.cc1175 ShapeHandle indices_shape; in SparseSegmentReductionShapeFn() local
1199 ShapeHandle indices_shape; in SparseSegmentReductionGradShapeFn() local
1237 ShapeHandle indices_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
H A Darray_ops.cc1187 ShapeHandle indices_shape = c->input(1); in __anon38bbb0e81702() local
1208 ShapeHandle indices_shape = c->input(1); in __anon38bbb0e81802() local
3135 ShapeHandle indices_shape; in ScatterNdTensorShape() local
3181 ShapeHandle indices_shape; in __anon38bbb0e84702() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtest_utils_test.cc262 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
H A Dvalue_inference_test.cc273 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
291 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
308 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h573 const RuntimeShape& indices_shape) { in GatherNdHelper()
605 const RuntimeShape& indices_shape, in GatherNd()
633 const RuntimeShape& indices_shape, in GatherNdString()
659 inline TfLiteStatus ScatterNd(const RuntimeShape& indices_shape, in ScatterNd()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_cost_analysis_test.cc1144 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
1174 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
1208 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
/aosp_15_r20/external/ComputeLibrary/tests/datasets/
H A DGatherDataset.h92 void add_config(TensorShape input_shape, TensorShape indices_shape, int axis) in add_config()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dsegment_reduction_ops.cc48 TensorShape indices_shape = ctx->InputShape(1); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_one_hot_op_test.py151 indices_shape, argument
H A Dragged_gather_op_test.py394 indices_shape, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/
H A Dmax_unpooling_2d.cc76 const RuntimeShape indices_shape = GetTensorShape(indices); in Prepare() local

123