Home
last modified time | relevance | path

Searched defs:start_indices (Results 1 – 25 of 54) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dgather_operation_test.cc33 Literal* start_indices) { in RunTest()
63 Literal start_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local
87 Literal start_indices = LiteralUtil::CreateR1<int32_t>({1, 0}); in XLA_TEST_F() local
108 Literal start_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local
129 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 2}, {2, 1}}); in XLA_TEST_F() local
150 Literal start_indices = in XLA_TEST_F() local
172 Literal start_indices = in XLA_TEST_F() local
196 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in XLA_TEST_F() local
219 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in XLA_TEST_F() local
240 Literal start_indices = LiteralUtil::CreateR1<int32_t>({1, 1}); in XLA_TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dgather_expander.cc32 HloInstruction* start_indices, int64_t index_vector_dim) { in TransposeIndexVectorDimToLast()
59 HloInstruction* start_indices, int64_t index_vector_dim) { in CanonicalizeGatherIndices()
159 HloInstruction* const start_indices = incoming_loop_state[1]; in GatherLoopBody() local
276 HloInstruction* start_indices = gather_instr->mutable_operand(1); in GatherLoopTripCount() local
355 HloInstruction* start_indices = gather_instr->mutable_operand(1); in ExpandInstruction() local
H A Dgather_simplifier.cc53 auto* start_indices = gather->operands()[1]; in ExpandInstruction() local
134 auto* start_indices = gather->operands()[1]; in InstructionMatchesPattern() local
H A Dhlo_creation_utils.cc103 absl::Span<const int64_t> start_indices, in MakeSliceHlo()
162 HloInstruction* operand, absl::Span<HloInstruction* const> start_indices, in MakeDynamicSliceHlo()
179 HloInstruction* operand, HloInstruction* start_indices, in MakeDynamicSliceHlo()
209 HloInstruction* start_indices, const OpMetadata* metadata) { in MakeDynamicUpdateSliceHlo()
H A Dhlo_evaluator_test.cc3347 Literal start_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local
3371 Literal start_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local
3395 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 2}, {2, 1}}); in TEST_F() local
3423 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in TEST_F() local
3450 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in TEST_F() local
3474 Literal start_indices = LiteralUtil::CreateR1<int32_t>({1, 1}); in TEST_F() local
3498 Literal start_indices = LiteralUtil::CreateR2<int32_t>({{2, 1}, {1, 1}}); in TEST_F() local
3521 Literal start_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local
3545 Literal start_indices = in TEST_F() local
H A Ddynamic_padder.cc867 std::vector<HloInstruction*> start_indices; in RewriteReverse() local
911 std::vector<HloInstruction*> start_indices(input->shape().rank(), zero_s32); in RewriteInputWithDynamicPadding() local
1270 std::vector<HloInstruction*> start_indices( in RewriteDynamicSelectAndScatterSamePadding() local
1473 std::vector<int64_t> start_indices(slice_shape.rank(), 0); in RewriteDynamicBinaryOp() local
H A Dspace_to_batch_converter.cc552 std::vector<int64_t> start_indices(rank, 0), in HaloDuplicateWithSlice() local
857 std::vector<int64_t> start_indices(rank, 0), in ChangeSpatialSizeOnSpaceToBatchedShape() local
2137 std::vector<int64_t> start_indices(rank, 0), in Propagate() local
2247 std::vector<int64_t> start_indices(previous_shape.rank(), 0), in Propagate() local
2420 std::vector<int64_t> start_indices(rank, 0), in BatchToSpace() local
H A Dcollective_decomposer_utils.cc44 std::vector<HloInstruction *> start_indices(shard_shape.rank(), zero); in CreateStartIndicesForCollectiveDecomposition() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtensor_list_utils.cc382 std::vector<xla::XlaOp> start_indices( in ExecuteTensorListPushBack() local
399 std::vector<xla::XlaOp> start_indices(element_shape.dimensions_size() + 1, in ExecuteTensorListPushBack() local
438 std::vector<xla::XlaOp> start_indices(list_part_shape.dimensions_size(), in ExecuteTensorListPopBack() local
486 std::vector<xla::XlaOp> start_indices(element_shape.dimensions_size() + 1, in ExecuteTensorListSetItem() local
519 std::vector<xla::XlaOp> start_indices(buffer_shape.dimensions_size(), in ExecuteTensorListGetItem() local
H A Dtensor_array_ops.cc125 absl::Span<const xla::XlaOp> start_indices, in DynamicAddSlice()
215 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
266 std::vector<xla::XlaOp> start_indices(ta_shape.dims(), in Compile() local
424 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
H A Dstack_ops.cc148 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
204 std::vector<xla::XlaOp> start_indices(stack_shape.dims(), in Compile() local
H A Dstrided_slice_op.cc145 std::vector<xla::XlaOp> start_indices; in EmitDynamicSlice() local
510 std::vector<int64_t> start_indices(input_shape.dims(), 0); in CompileAsDynamicUpdateSlice() local
H A Dunpack_op.cc67 std::vector<int64_t> start_indices(input_shape.dims(), 0); in Compile() local
H A Droll_op.cc83 std::vector<xla::XlaOp> start_indices( in Compile() local
H A Dreverse_sequence_op.cc97 xla::XlaOp start_indices = in Compile() local
H A Dfft_ops.cc87 std::vector<int64_t> start_indices(input_shape.dims(), 0); in Compile() local
H A Dbatchtospace_op.cc125 std::vector<int64_t> start_indices(input_rank, 0); in BatchToSpace() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dsorting.cc76 std::vector<int64_t> start_indices(input_shape.dimensions_size(), 0); in TopK() local
203 std::vector<int64_t> start_indices(input_shape.dimensions_size(), 0); in TopKWithPartitions() local
218 std::vector<int64_t> start_indices(input_shape.dimensions_size(), 0); in TopKWithPartitions() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dstrided_slice_logic.h78 const auto* start_indices = params.start_indices; in StartForAxis() local
184 const std::vector<int>& start_indices, const std::vector<int>& stop_indices, in BuildStridedSliceParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Dmath_utils_test.cc36 Eigen::DSizes<Eigen::Index, 3> start_indices{1, 0, 0}; in TEST() local
61 Eigen::DSizes<Eigen::Index, 3> start_indices{1, 0, 0}; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddynamic_update_slice.cc49 const TfLiteTensor* start_indices; in Prepare() local
84 const int* start_indices = nullptr) { in TensorIndexToFlat()
H A Ddynamic_update_slice_test.cc37 const TensorData& start_indices) { in DynamicUpdateSliceOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc382 const Shape& shape, XlaOp input, XlaOp start_indices, in GatherInternal()
653 const Shape& shape, XlaOp operand, absl::Span<const int64_t> start_indices, in SliceInternal()
663 const Shape& shape, XlaOp operand, absl::Span<const XlaOp> start_indices, in DynamicSliceInternal()
675 absl::Span<const XlaOp> start_indices) { in DynamicUpdateSliceInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_padding_legalization.cc96 std::vector<int64_t> start_indices(input->shape().dimensions_size(), 0); in MaybePaddedAndSlicedInput() local
354 std::vector<int64_t> start_indices( in CanonicalizeBackwardInputConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1018 XlaOp XlaBuilder::Slice(XlaOp operand, absl::Span<const int64_t> start_indices, in Slice()
1031 const Shape& shape, XlaOp operand, absl::Span<const int64_t> start_indices, in SliceInternal()
1062 absl::Span<const XlaOp> start_indices, in DynamicSlice()
1080 const Shape& shape, XlaOp operand, absl::Span<const XlaOp> start_indices, in DynamicSliceInternal()
1095 absl::Span<const XlaOp> start_indices) { in DynamicUpdateSlice()
1114 absl::Span<const XlaOp> start_indices) { in DynamicUpdateSliceInternal()
2395 XlaOp XlaBuilder::Gather(XlaOp input, XlaOp start_indices, in Gather()
2412 const Shape& shape, XlaOp input, XlaOp start_indices, in GatherInternal()
4110 XlaOp Slice(const XlaOp operand, absl::Span<const int64_t> start_indices, in Slice()
4123 XlaOp DynamicSlice(const XlaOp operand, absl::Span<const XlaOp> start_indices, in DynamicSlice()
[all …]

123