| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op.h | 34 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
42 start_i[i] = start_indices[i];
50 input.stridedSlice(start_indices, stop_indices, strides);
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) =
|
| fft_ops.cc | 201 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> start_indices; variable
205 full_fft.slice(start_indices, input_slice_sizes).device(device) =
206 input.slice(start_indices, input_slice_sizes);
214 full_fft.slice(start_indices, input_slice_sizes).device(device) =
215 full_fft.slice(start_indices, input_slice_sizes)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_strided_slice_attributes.cc | 28 if (!op->start_indices.empty()) {
51 op->start_indices = start_array.GetBuffer<ArrayDataType::kInt32>().data;
55 CHECK_GE(op->start_indices.size(), 1);
56 CHECK_LE(op->start_indices.size(), 4);
57 CHECK_EQ(op->stop_indices.size(), op->start_indices.size());
|
| resolve_constant_strided_slice.cc | 34 start = op.start_indices[axis];
73 int num_input_axes = op.start_indices.size();
88 std::vector<int> src_coord(op.start_indices.size());
149 if (op->start_indices.empty() || op->stop_indices.empty() ||
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| unpack_op.cc | 67 std::vector<int64> start_indices(input_shape.dims(), 0);
75 start_indices[axis] = i;
77 auto slice = ctx->builder()->Slice(input, start_indices, limit_indices,
|
| stack_ops.cc | 154 // start_indices of the DynamicUpdateSlice are [index, 0, 0, ..., 0].
155 auto start_indices = variable
166 {b->DynamicUpdateSlice(ta, update, start_indices),
209 // start_indices of the DynamicSlice are [index, 0, 0, ..., 0].
210 auto start_indices = variable
220 b->DynamicSlice(ta, start_indices, slice_shape);
|
| batchtospace_op.cc | 126 std::vector<int64> start_indices(input_rank, 0);
134 start_indices[1 + i] = crop_start;
137 ctx, start_indices[1 + i] <= end_indices[1 + i],
143 b->Slice(reshaped_permuted, start_indices, end_indices, strides);
|
| tensor_array_ops.cc | 130 const xla::ComputationDataHandle& start_indices) {
132 builder->DynamicSlice(operand, start_indices, update_dims);
134 return builder->DynamicUpdateSlice(operand, sum, start_indices);
221 // start_indices of the DynamicUpdateSlice are [index, 0, 0, ..., 0].
222 auto start_indices = variable
231 DynamicAddSlice(b, ta, update, slice_shape.dim_sizes(), start_indices);
265 // start_indices of the DynamicSlice are [index, 0, 0, ..., 0].
266 auto start_indices = variable
274 b->DynamicSlice(ta, start_indices, slice_shape);
418 // start_indices of the DynamicUpdateSlice are [index, 0, 0, ..., 0]
420 auto start_indices = variable
[all...] |
| gather_op.cc | 134 auto start_indices = bodyb->Pad(
139 bodyb->DynamicSlice(input, start_indices, slice_shape.dim_sizes()),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| pad_insertion.cc | 86 std::vector<int64> start_indices(input->shape().dimensions_size(), 0);
94 start_indices[dim] +=
101 ShapeInference::InferSliceShape(input->shape(), start_indices,
104 input, start_indices, limit_indices, strides));
345 // Initialize start_indices and limit_indices as no slicing.
346 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(),
360 start_indices[dim] += padding_low - padding_high;
370 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices,
379 start_indices, limit_indices, strides));
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ops.cc | 128 const auto* start_indices = dynamic_update_slice->operand(2); local
147 // Create element generators for update and start_indices.
152 fused_emitter.GetGenerator(start_indices);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 365 tensorflow::gtl::ArraySlice<int64> start_indices,
368 return builder_.Slice(operand, start_indices, limit_indices, strides);
373 const ComputationDataHandle& start_indices,
375 return builder_.DynamicSlice(operand, start_indices, slice_sizes);
380 const ComputationDataHandle& start_indices) {
381 return builder_.DynamicUpdateSlice(operand, update, start_indices);
|
| local_computation_builder.h | 169 tensorflow::gtl::ArraySlice<int64> start_indices,
175 const ComputationDataHandle& start_indices,
180 const ComputationDataHandle& start_indices);
|
| xla_client.py | 755 def Slice(self, operand, start_indices, limit_indices, strides=None):
760 start_indices: iterable of N integers containing the starting indices of
771 start_indices = list(start_indices)
772 strides = [1] * len(start_indices)
776 start_indices,
780 def DynamicSlice(self, operand, start_indices, slice_sizes):
785 start_indices: ComputationDataHandle for the 1D array of N integers
796 _unwrap_data_handle(start_indices),
799 def DynamicUpdateSlice(self, operand, update, start_indices)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| test_utils.cc | 177 std::vector<int32> start_indices(rank);
182 start_indices[i] = generator(*engine);
184 return Literal::CreateR1<int32>(start_indices);
|
| dynamic_ops_test.cc | 685 auto start_indices = local
686 builder.Parameter(0, start_indices_shape, "start_indices");
688 builder.DynamicSlice(input, start_indices, {1, 1, 1, 1});
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_rematerialization_test.cc | 69 vec1_shape_, concat_1, /*start_indices=*/{0},
78 /*start_indices=*/{0},
107 HloInstruction::CreateSlice(vec1_shape_, bcast, /*start_indices=*/{0},
116 /*start_indices=*/{0},
429 vec1024_shape_, concat, /*start_indices=*/{0},
545 vec1024_shape_, concat, /*start_indices=*/{0},
|
| user_computation.cc | 2930 HloInstruction* start_indices = local
2946 HloInstruction* start_indices = local
[all...] |
| algebraic_simplifier.cc | 877 std::array<int64, 2> start_indices; local
1306 std::vector<int64> start_indices; local
1576 auto start_indices = dynamic_slice->operand(1); local
1591 auto start_indices = dynamic_update_slice->operand(2); local
[all...] |
| hlo_evaluator.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.h | 240 tensorflow::gtl::ArraySlice<int64> start_indices,
255 // from dynamic start indices which are passed in 'start_indices'.
259 // The shape of 'start_indices' must be rank == 1, with dimension size
265 const ComputationDataHandle& start_indices,
269 // updates a slice of 'operand' with 'update' at dynamic 'start_indices'.
272 // The indices specified in 'start_indices' specify the offset of the slice
280 // The shape of 'start_indices' must be rank == 1, with dimension size
286 const ComputationDataHandle& start_indices);
[all...] |
| computation_builder.cc | 278 tensorflow::gtl::ArraySlice<int64> start_indices,
284 for (int64 index : start_indices) {
317 const ComputationDataHandle& start_indices,
322 *request->mutable_start_indices() = start_indices;
331 const ComputationDataHandle& start_indices) {
337 *request->mutable_start_indices() = start_indices;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| array.h | 439 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)) {
|
| xla_data.proto | 437 repeated int64 start_indices = 3;
443 // Operand from which to slice at dynamic 'start_indices'.
445 // Dynamically computed 'start_indices' for slice operation.
446 ComputationDataHandle start_indices = 3;
458 ComputationDataHandle start_indices = 4;
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| operator_test.cc | 372 EXPECT_EQ(op.start_indices, output_toco_op->start_indices);
|
