| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| unpack_op.cc | 68 std::vector<int64> limit_indices(input_shape.dims());
71 limit_indices[i] = input_shape.dim_size(i);
76 limit_indices[axis] = i + 1;
77 auto slice = ctx->builder()->Slice(input, start_indices, limit_indices,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| pad_insertion.cc | 87 std::vector<int64> limit_indices(input->shape().dimensions().begin(),
96 limit_indices[dim] -=
102 limit_indices, strides)
104 input, start_indices, limit_indices, strides));
345 // Initialize start_indices and limit_indices as no slicing.
348 std::vector<int64> limit_indices(
364 limit_indices[dim] -= padding_high - padding_low;
371 limit_indices, strides)
379 start_indices, limit_indices, strides));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| array.h | 441 std::vector<int64> limit_indices;
443 from.dimensions().begin(), std::back_inserter(limit_indices),
448 if (array_impl::all_inside_range(index, start_indices, limit_indices)) {
|
| xla_data.proto | 438 repeated int64 limit_indices = 4;
|
| literal_util.cc | 805 tensorflow::gtl::ArraySlice<int64> limit_indices) const {
811 CHECK_LE(limit_indices[dnum], shape().dimensions(dnum));
812 int64 dimension = limit_indices[dnum] - start_indices[dnum];
[all...] |
| literal_util.h | 317 // start_indices and limit_indices must be the rank of the literal, and the
322 tensorflow::gtl::ArraySlice<int64> limit_indices) const;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 366 tensorflow::gtl::ArraySlice<int64> limit_indices,
368 return builder_.Slice(operand, start_indices, limit_indices, strides);
|
| local_computation_builder.h | 170 tensorflow::gtl::ArraySlice<int64> limit_indices,
|
| xla_client.py | 755 def Slice(self, operand, start_indices, limit_indices, strides=None):
762 limit_indices: iterable of N integers containing the ending indices
777 limit_indices,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier.cc | 881 std::array<int64, 2> limit_indices; local
[all...] |
| hlo_instruction.h | 338 tensorflow::gtl::ArraySlice<int64> limit_indices,
[all...] |
| hlo_instruction.cc | 502 tensorflow::gtl::ArraySlice<int64> limit_indices,
507 instruction->slice_limits_.assign(limit_indices.begin(), limit_indices.end());
[all...] |
| user_computation.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 279 tensorflow::gtl::ArraySlice<int64> limit_indices,
287 for (int64 index : limit_indices) {
[all...] |
| computation_builder.h | 241 tensorflow::gtl::ArraySlice<int64> limit_indices,
[all...] |
