Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/tests/
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 …]
/external/tensorflow/tensorflow/compiler/xla/service/
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
Dgather_simplifier.cc53 auto* start_indices = gather->operands()[1]; in ExpandInstruction() local
134 auto* start_indices = gather->operands()[1]; in InstructionMatchesPattern() local
Dcollective_decomposer_utils.cc44 std::vector<HloInstruction *> start_indices(shard_shape.rank(), zero); in CreateStartIndicesForCollectiveDecomposition() local
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()
Ddot_merger.cc197 DimensionVector start_indices(new_dot_shape.dimensions_size(), 0); in TryMergeSameOperand() local
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
/external/tensorflow/tensorflow/compiler/xla/client/lib/
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
/external/tensorflow/tensorflow/lite/kernels/internal/
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()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
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
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
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
Dunpack_op.cc67 std::vector<int64_t> start_indices(input_shape.dims(), 0); in Compile() local
Droll_op.cc83 std::vector<xla::XlaOp> start_indices( in Compile() local
Dreverse_sequence_op.cc97 xla::XlaOp start_indices = in Compile() local
Dfft_ops.cc87 std::vector<int64_t> start_indices(input_shape.dims(), 0); in Compile() local
Dbatchtospace_op.cc125 std::vector<int64_t> start_indices(input_rank, 0); in BatchToSpace() local
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
/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
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
/external/tensorflow/tensorflow/lite/kernels/
Ddynamic_update_slice.cc49 const TfLiteTensor* start_indices; in Prepare() local
84 const int* start_indices = nullptr) { in TensorIndexToFlat()
Ddynamic_update_slice_test.cc37 const TensorData& start_indices) { in DynamicUpdateSliceOpModel()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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
Dcublas_pad_for_gemms.cc100 std::vector<int64_t> start_indices(result_shape.rank(), 0); in PadForGemm() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
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()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc200 const auto* start_indices = dynamic_update_slice->operand(2); in EmitFusedDynamicUpdateSliceInPlaceImpl() local

123