Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dgather_operation_test.cc33 Literal* start_indices) { in RunTest()
63 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
84 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
105 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in XLA_TEST_F() local
126 Literal start_indices = in XLA_TEST_F() local
148 Literal start_indices = in XLA_TEST_F() local
172 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F() local
195 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F() local
216 Literal start_indices = LiteralUtil::CreateR1<int32>({1, 1}); in XLA_TEST_F() local
237 Literal start_indices = LiteralUtil::CreateR2<int32>({{2, 1}, {1, 1}}); in XLA_TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc30 HloInstruction* start_indices, int64 index_vector_dim) { in TransposeIndexVectorDimToLast()
57 HloInstruction* start_indices, int64 index_vector_dim) { in CanonicalizeGatherIndices()
156 HloInstruction* const start_indices = incoming_loop_state[1]; in GatherLoopBody() local
313 HloInstruction* start_indices = gather_instr->mutable_operand(1); in ExpandInstruction() local
Dhlo_creation_utils.cc71 absl::Span<const int64> start_indices, in MakeSliceHlo()
122 HloInstruction* operand, HloInstruction* start_indices, in MakeDynamicSliceHlo()
150 HloInstruction* start_indices) { in MakeDynamicUpdateSliceHlo()
Dhlo_evaluator_test.cc2106 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2130 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2154 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in TEST_F() local
2182 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
2209 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
2233 Literal start_indices = LiteralUtil::CreateR1<int32>({1, 1}); in TEST_F() local
2257 Literal start_indices = LiteralUtil::CreateR2<int32>({{2, 1}, {1, 1}}); in TEST_F() local
2280 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2304 Literal start_indices = in TEST_F() local
Dhlo_evaluator_typed_visitor.h1458 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice() local
1521 auto start_indices = dynamic_update_slice->operand(2); in HandleDynamicUpdateSlice() local
2860 absl::Span<HloInstruction* const> start_indices, in DynamicSlice()
2895 absl::Span<HloInstruction* const> start_indices) { in DynamicUpdateSlice()
Dtriangular_solve_expander.cc61 auto start_indices = in DiagonalBlocks() local
Dhlo_evaluator.cc832 const Shape& output_shape, const Literal* start_indices) in OutputBatchIndexToInputIndex()
1033 int64 index_vector_dim, const Literal& start_indices, in ReshapedGatherIndices()
1531 std::vector<int64> start_indices(rank, 0); in HandleSort() local
Dhlo_instructions.cc965 absl::Span<const int64> start_indices, in HloSliceInstruction()
2234 const Shape& shape, HloInstruction* operand, HloInstruction* start_indices, in HloDynamicSliceInstruction()
2244 absl::Span<HloInstruction* const> start_indices, in HloDynamicSliceInstruction()
2256 HloInstruction* start_indices) in HloDynamicUpdateSliceInstruction()
2265 absl::Span<HloInstruction* const> start_indices) in HloDynamicUpdateSliceInstruction()
2310 const Shape& shape, HloInstruction* operand, HloInstruction* start_indices, in HloGatherInstruction()
Dhlo_instruction.cc986 absl::Span<const int64> start_indices, in CreateSlice()
994 absl::Span<HloInstruction* const> start_indices, in CreateDynamicSlice()
1003 absl::Span<HloInstruction* const> start_indices) { in CreateDynamicUpdateSlice()
1307 const Shape& shape, HloInstruction* operand, HloInstruction* start_indices, in CreateGather()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h77 const auto* start_indices = params.start_indices; in StartForAxis() local
171 const std::vector<int>& start_indices, const std::vector<int>& stop_indices, in BuildStridedSliceParams()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc41 std::vector<int64> start_indices(input_shape.dimensions_size(), 0); in TopK() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc224 std::vector<xla::XlaOp> start_indices(shape.dims(), in Compile() local
362 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
405 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
450 std::vector<xla::XlaOp> start_indices(shape.dims(), in Compile() 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
Dunpack_op.cc67 std::vector<int64> start_indices(input_shape.dims(), 0); 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
423 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile() local
Dreverse_sequence_op.cc96 xla::XlaOp start_indices = in Compile() local
Dfft_ops.cc85 std::vector<int64> start_indices(input_shape.dims(), 0); in Compile() local
Dbatchtospace_op.cc125 std::vector<int64> start_indices(input_rank, 0); in BatchToSpace() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc94 std::vector<int64> start_indices(input->shape().dimensions_size(), 0); in MaybePaddedAndSlicedInput() local
347 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(), in CanonicalizeBackwardInputConvolution() local
Dcudnn_conv_pad_for_tensor_cores.cc101 std::vector<int64> start_indices(result_shape.dimensions_size(), 0); in PadConv() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc516 absl::Span<const int64> start_indices, in Slice()
529 const LocalOp& start_indices, in DynamicSlice()
536 const LocalOp& start_indices) { in DynamicUpdateSlice()
755 const LocalOp& input, const LocalOp& start_indices, in Gather()
Dxla_client.py1317 def Slice(self, operand, start_indices, limit_indices, strides=None): argument
1353 def DynamicSlice(self, operand, start_indices, slice_sizes): argument
1368 def DynamicUpdateSlice(self, operand, update, start_indices): argument
1775 def Gather(self, a, start_indices, dimension_numbers, slice_sizes): argument
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc149 const auto* start_indices = dynamic_update_slice->operand(2); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc722 absl::Span<const int64> start_indices, in Slice()
758 XlaOp XlaBuilder::DynamicSlice(const XlaOp& operand, const XlaOp& start_indices, in DynamicSlice()
781 absl::Span<const XlaOp> start_indices, in DynamicSlice()
809 const XlaOp& start_indices) { in DynamicUpdateSlice()
828 absl::Span<const XlaOp> start_indices) { in DynamicUpdateSlice()
1840 XlaOp XlaBuilder::Gather(const XlaOp& input, const XlaOp& start_indices, in Gather()
2868 XlaOp Slice(const XlaOp& operand, absl::Span<const int64> start_indices, in Slice()
2881 XlaOp DynamicSlice(const XlaOp& operand, const XlaOp& start_indices, in DynamicSlice()
2885 XlaOp DynamicSlice(const XlaOp& operand, absl::Span<const XlaOp> start_indices, in DynamicSlice()
2891 const XlaOp& start_indices) { in DynamicUpdateSlice()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc211 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> start_indices; in DoFFT() local

12