| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unpack_op.cc | 67 std::vector<int64> start_indices(input_shape.dims(), 0); in Compile() local
|
| D | stack_ops.cc | 155 auto start_indices = in Compile() local
210 auto start_indices = in Compile() local
|
| D | tensor_array_ops.cc | 130 const xla::ComputationDataHandle& start_indices) { in DynamicAddSlice()
222 auto start_indices = in Compile() local
266 auto start_indices = in Compile() local
420 auto start_indices = in Compile() local
|
| D | batchtospace_op.cc | 126 std::vector<int64> start_indices(input_rank, 0); in BatchToSpace() local
|
| D | gather_op.cc | 134 auto start_indices = bodyb->Pad( in XlaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | pad_insertion.cc | 86 std::vector<int64> start_indices(input->shape().dimensions_size(), 0); in MaybePaddedAndSlicedInput() local
346 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(), in CanonicalizeBackwardInputConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 365 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice()
373 const ComputationDataHandle& start_indices, in DynamicSlice()
380 const ComputationDataHandle& start_indices) { in DynamicUpdateSlice()
|
| D | xla_client.py | 755 def Slice(self, operand, start_indices, limit_indices, strides=None): argument
780 def DynamicSlice(self, operand, start_indices, slice_sizes): argument
799 def DynamicUpdateSlice(self, operand, update, start_indices): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ops.cc | 128 const auto* start_indices = dynamic_update_slice->operand(2); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 877 std::array<int64, 2> start_indices; in OptimizeDotOfConcatHelper() local
1306 std::vector<int64> start_indices; in HandlePad() local
1576 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice() local
1591 auto start_indices = dynamic_update_slice->operand(2); in HandleDynamicUpdateSlice() local
|
| D | hlo_evaluator_test.cc | 1469 auto start_indices = b.AddInstruction( in TEST_P() local
1504 auto start_indices = b.AddInstruction( in TEST_P() local
1537 auto start_indices = b.AddInstruction( in TEST_P() local
|
| D | hlo_instruction.cc | 501 tensorflow::gtl::ArraySlice<int64> start_indices, in CreateSlice()
519 const Shape& shape, HloInstruction* operand, HloInstruction* start_indices, in CreateDynamicSlice()
534 HloInstruction* start_indices) { in CreateDynamicUpdateSlice()
|
| D | user_computation.cc | 2930 HloInstruction* start_indices = in Visit() local
2946 HloInstruction* start_indices = in Visit() local
|
| D | hlo_evaluator.cc | 1231 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice() local
1286 auto start_indices = dynamic_update_slice->operand(2); in HandleDynamicUpdateSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 177 std::vector<int32> start_indices(rank); in MakeRandomNonwrappingSliceIndex() local
|
| D | dynamic_ops_test.cc | 685 auto start_indices = in BM_DynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 437 repeated int64 start_indices = 3; field
446 ComputationDataHandle start_indices = 3; field
458 ComputationDataHandle start_indices = 4; field
|
| D | literal_util.cc | 804 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 201 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> start_indices; in DoFFT() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | computation_builder.cc | 278 tensorflow::gtl::ArraySlice<int64> start_indices, in Slice()
317 const ComputationDataHandle& start_indices, in DynamicSlice()
331 const ComputationDataHandle& start_indices) { in DynamicUpdateSlice()
|