Home
last modified time | relevance | path

Searched refs:start_indices (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.h34 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices, in operator()
42 start_i[i] = start_indices[i]; in operator()
50 input.stridedSlice(start_indices, stop_indices, strides); in operator()
82 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
91 start_i[i] = start_indices[i];
98 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
108 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
116 start_i[i] = start_indices[i];
123 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
Dfft_ops.cc201 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> start_indices; in DoFFT() local
205 full_fft.slice(start_indices, input_slice_sizes).device(device) = in DoFFT()
206 input.slice(start_indices, input_slice_sizes); in DoFFT()
214 full_fft.slice(start_indices, input_slice_sizes).device(device) = in DoFFT()
215 full_fft.slice(start_indices, input_slice_sizes) in DoFFT()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc28 if (!op->start_indices.empty()) { in Run()
51 op->start_indices = start_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
55 CHECK_GE(op->start_indices.size(), 1); in Run()
56 CHECK_LE(op->start_indices.size(), 4); in Run()
57 CHECK_EQ(op->stop_indices.size(), op->start_indices.size()); in Run()
Dresolve_constant_strided_slice.cc34 start = op.start_indices[axis]; in StartForAxis()
73 int num_input_axes = op.start_indices.size(); in StridedSlice()
88 std::vector<int> src_coord(op.start_indices.size()); in StridedSlice()
149 if (op->start_indices.empty() || op->stop_indices.empty() || in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpad_insertion.cc86 std::vector<int64> start_indices(input->shape().dimensions_size(), 0); in MaybePaddedAndSlicedInput() local
94 start_indices[dim] += in MaybePaddedAndSlicedInput()
101 ShapeInference::InferSliceShape(input->shape(), start_indices, in MaybePaddedAndSlicedInput()
104 input, start_indices, limit_indices, strides)); in MaybePaddedAndSlicedInput()
346 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(), in CanonicalizeBackwardInputConvolution() local
360 start_indices[dim] += padding_low - padding_high; in CanonicalizeBackwardInputConvolution()
370 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices, in CanonicalizeBackwardInputConvolution()
379 start_indices, limit_indices, strides)); in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunpack_op.cc67 std::vector<int64> start_indices(input_shape.dims(), 0); in Compile() local
75 start_indices[axis] = i; in Compile()
77 auto slice = ctx->builder()->Slice(input, start_indices, limit_indices, in Compile()
Dbatchtospace_op.cc126 std::vector<int64> start_indices(input_rank, 0); in BatchToSpace() local
134 start_indices[1 + i] = crop_start; in BatchToSpace()
137 ctx, start_indices[1 + i] <= end_indices[1 + i], in BatchToSpace()
143 b->Slice(reshaped_permuted, start_indices, end_indices, strides); in BatchToSpace()
Dtensor_array_ops.cc130 const xla::ComputationDataHandle& start_indices) { in DynamicAddSlice() argument
132 builder->DynamicSlice(operand, start_indices, update_dims); in DynamicAddSlice()
134 return builder->DynamicUpdateSlice(operand, sum, start_indices); in DynamicAddSlice()
222 auto start_indices = in Compile() local
231 DynamicAddSlice(b, ta, update, slice_shape.dim_sizes(), start_indices); in Compile()
266 auto start_indices = in Compile() local
274 b->DynamicSlice(ta, start_indices, slice_shape); in Compile()
420 auto start_indices = in Compile() local
423 ta = DynamicAddSlice(b, ta, slice, slice_dims, start_indices); in Compile()
Dstack_ops.cc155 auto start_indices = in Compile() local
166 {b->DynamicUpdateSlice(ta, update, start_indices), in Compile()
210 auto start_indices = in Compile() local
220 b->DynamicSlice(ta, start_indices, slice_shape); in Compile()
Dgather_op.cc134 auto start_indices = bodyb->Pad( in XlaGather() local
139 bodyb->DynamicSlice(input, start_indices, slice_shape.dim_sizes()), in XlaGather()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py755 def Slice(self, operand, start_indices, limit_indices, strides=None): argument
771 start_indices = list(start_indices)
772 strides = [1] * len(start_indices)
776 start_indices,
780 def DynamicSlice(self, operand, start_indices, slice_sizes): argument
796 _unwrap_data_handle(start_indices),
799 def DynamicUpdateSlice(self, operand, update, start_indices): argument
814 _unwrap_data_handle(start_indices)))
Dlocal_computation_builder.cc365 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice() argument
368 return builder_.Slice(operand, start_indices, limit_indices, strides); in Slice()
373 const ComputationDataHandle& start_indices, in DynamicSlice() argument
375 return builder_.DynamicSlice(operand, start_indices, slice_sizes); in DynamicSlice()
380 const ComputationDataHandle& start_indices) { in DynamicUpdateSlice() argument
381 return builder_.DynamicUpdateSlice(operand, update, start_indices); in DynamicUpdateSlice()
Dlocal_computation_builder.h169 tensorflow::gtl::ArraySlice<int64> start_indices,
175 const ComputationDataHandle& start_indices,
180 const ComputationDataHandle& start_indices);
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc177 std::vector<int32> start_indices(rank); in MakeRandomNonwrappingSliceIndex() local
182 start_indices[i] = generator(*engine); in MakeRandomNonwrappingSliceIndex()
184 return Literal::CreateR1<int32>(start_indices); in MakeRandomNonwrappingSliceIndex()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dops.cc128 const auto* start_indices = dynamic_update_slice->operand(2); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
152 fused_emitter.GetGenerator(start_indices); in EmitFusedDynamicUpdateSliceInPlaceImpl()
/external/tensorflow/tensorflow/compiler/xla/
Darray.h439 tensorflow::gtl::ArraySlice<int64> start_indices) {
442 std::transform(start_indices.begin(), start_indices.end(),
448 if (array_impl::all_inside_range(index, start_indices, limit_indices)) {
Dliteral_util.cc804 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice() argument
810 CHECK_GE(start_indices[dnum], 0); in Slice()
812 int64 dimension = limit_indices[dnum] - start_indices[dnum]; in Slice()
828 new_indices[i] = indices[i] + start_indices[i]; in Slice()
838 new_indices[i] = indices[i] + start_indices[i]; in Slice()
848 new_indices[i] = indices[i] + start_indices[i]; in Slice()
858 new_indices[i] = indices[i] + start_indices[i]; in Slice()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc877 std::array<int64, 2> start_indices; in OptimizeDotOfConcatHelper() local
878 start_indices[rhs_contracting_dim] = rhs_contracting_dim_offset; in OptimizeDotOfConcatHelper()
879 start_indices[1 - rhs_contracting_dim] = 0; in OptimizeDotOfConcatHelper()
887 rhs_slice_shape, rhs, /*start_indices=*/start_indices, in OptimizeDotOfConcatHelper()
1306 std::vector<int64> start_indices; in HandlePad() local
1320 start_indices.push_back(start); in HandlePad()
1328 ShapeInference::InferSliceShape(nonzero_pad_shape, start_indices, in HandlePad()
1333 pad->shape(), nonzero_pad, start_indices, end_indices, strides); in HandlePad()
1576 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice() local
1582 if (IsAll(start_indices, 0) && SameShape(operand, dynamic_slice)) { in HandleDynamicSlice()
[all …]
Duser_computation.cc926 operand->output_shape(), AsInt64Slice(slice_request.start_indices()), in AddSliceInstruction()
951 TF_ASSIGN_OR_RETURN(const OperationRequest* start_indices, in AddDynamicSliceInstruction()
952 LookUpRequest(dynamic_slice_request.start_indices())); in AddDynamicSliceInstruction()
957 operand->output_shape(), start_indices->output_shape(), in AddDynamicSliceInstruction()
987 const OperationRequest* start_indices, in AddDynamicUpdateSliceInstruction()
988 LookUpRequest(dynamic_update_slice_request.start_indices())); in AddDynamicUpdateSliceInstruction()
993 start_indices->output_shape())); in AddDynamicUpdateSliceInstruction()
1704 dynamic_slice_request.start_indices(), in PureFunctionalVisitor()
1719 dynamic_update_slice_request.start_indices(), in PureFunctionalVisitor()
2488 apply(dynamic_slice_request.start_indices()); in ForEachOperand()
[all …]
Dhlo_evaluator.cc1231 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice() local
1235 operand->shape(), start_indices->shape(), in HandleDynamicSlice()
1242 primitive_util::IsIntegralType(start_indices->shape().element_type())); in HandleDynamicSlice()
1246 parent_->GetEvaluatedLiteralFor(start_indices); in HandleDynamicSlice()
1248 switch (start_indices->shape().element_type()) { in HandleDynamicSlice()
1276 << PrimitiveType_Name(start_indices->shape().element_type()); in HandleDynamicSlice()
1286 auto start_indices = dynamic_update_slice->operand(2); in HandleDynamicUpdateSlice() local
1291 operand->shape(), update->shape(), start_indices->shape())); in HandleDynamicUpdateSlice()
1297 primitive_util::IsIntegralType(start_indices->shape().element_type())); in HandleDynamicUpdateSlice()
1303 parent_->GetEvaluatedLiteralFor(start_indices); in HandleDynamicUpdateSlice()
[all …]
Dhlo_evaluator_test.cc1469 auto start_indices = b.AddInstruction( in TEST_P() local
1474 start_indices, {2, 3})); in TEST_P()
1504 auto start_indices = b.AddInstruction( in TEST_P() local
1509 start_indices, {2, 3})); in TEST_P()
1537 auto start_indices = b.AddInstruction( in TEST_P() local
1545 shape, operand, update, start_indices)); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/client/
Dcomputation_builder.cc278 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice() argument
284 for (int64 index : start_indices) { in Slice()
317 const ComputationDataHandle& start_indices, in DynamicSlice() argument
322 *request->mutable_start_indices() = start_indices; in DynamicSlice()
331 const ComputationDataHandle& start_indices) { in DynamicUpdateSlice() argument
337 *request->mutable_start_indices() = start_indices; in DynamicUpdateSlice()
Dcomputation_builder.h240 tensorflow::gtl::ArraySlice<int64> start_indices,
265 const ComputationDataHandle& start_indices,
286 const ComputationDataHandle& start_indices);
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc372 EXPECT_EQ(op.start_indices, output_toco_op->start_indices); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser_test.cc529 ENTRY %DynamicUpdateSlice.v4 (input: s32[1,1,25,1], update: s32[1,1,2,1], start_indices: s32[4]) ->…
532 %start_indices = s32[4]{0} parameter(2)
533 …pdate-slice(s32[1,1,25,1]{3,2,1,0} %input, s32[1,1,2,1]{3,2,1,0} %update, s32[4]{0} %start_indices)

12