| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 381 std::array<int64, 2> slice_strides; member
399 Slice(a, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
404 input, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
466 std::array<int64, 4> slice_strides; member
475 "__strides_", absl::StrJoin(spec.slice_strides, "x")); in R4SpecToString()
486 values, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
493 Slice(parameter, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 76 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
82 slice_strides.push_back(strides[i]); in Compile()
89 slice_strides.push_back(-strides[i]); in Compile()
99 slice = xla::Slice(slice, slice_begin, slice_end, slice_strides); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 476 std::vector<int64> slice_strides(filter->shape().dimensions_size(), 1); in HandleConvolution() local
505 filter_slice_limits, slice_strides)); in HandleConvolution()
515 activation_slice_limits, slice_strides)); in HandleConvolution()
|
| D | hlo_constant_folding_test.cc | 155 const int64 slice_strides[] = {1, 1, 1, 1, 1}; in TEST_F() local
163 shape, literal_instruction, slice_start, slice_limits, slice_strides)); in TEST_F()
|
| D | hlo_instruction.cc | 289 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
294 slice_strides.push_back(slice_dimensions.stride()); in CreateFromProto()
297 slice_strides); in CreateFromProto()
3263 int64 HloInstruction::slice_strides(int64 dimension) const { in slice_strides() function in xla::HloInstruction
3264 return Cast<HloSliceInstruction>(this)->slice_strides(dimension); in slice_strides()
3267 const std::vector<int64>& HloInstruction::slice_strides() const { in slice_strides() function in xla::HloInstruction
3268 return Cast<HloSliceInstruction>(this)->slice_strides(); in slice_strides()
|
| D | hlo_instruction.h | 1418 int64 slice_strides(int64 dimension) const;
1419 const std::vector<int64>& slice_strides() const;
|
| D | dynamic_dimension_inference.cc | 399 hlo->slice_strides(dimension) != 1 || in HandleSlice()
|
| D | algebraic_simplifier.cc | 161 return absl::c_all_of(hlo->slice_strides(), in IsUnstridedSlice()
723 /*strides=*/operands[i]->slice_strides())); in HandleConcatenate()
2759 slice->slice_strides()))); in TrySimplifyScalarSlice()
2838 new_slice_starts, new_slice_limits, slice->slice_strides())); in HandleSlice()
2846 if (slice->slice_starts(dim) != 0 || slice->slice_strides(dim) != 1 || in HandleSlice()
|
| D | hlo_instructions.h | 627 int64 slice_strides(int64 dimension) const { in slice_strides() function
630 const std::vector<int64>& slice_strides() const { return slice_strides_; } in slice_strides() function
|
| D | hlo_verifier.cc | 572 slice->slice_limits(), slice->slice_strides())); in HandleSlice()
|
| D | hlo_evaluator_typed_visitor.h | 2420 slice->slice_limits(), slice->slice_strides())); in HandleSlice()
2432 slice->slice_starts(i) + out_index[i] * slice->slice_strides(i); in HandleSlice()
|
| D | elemental_ir_emitter.cc | 2371 /*strides=*/hlo->slice_strides(), /*builder=*/b_); in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3156 NodeDef* slice_strides; in GetSliceAxis() local
3159 TF_RETURN_IF_ERROR(GetInputNode(node->input(3), &slice_strides)); in GetSliceAxis()
3161 for (const auto* n : {slice_begin, slice_end, slice_strides}) { in GetSliceAxis()
3177 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_strides, "value")); in GetSliceAxis()
3179 slice_strides->attr().at("value").tensor())) { in GetSliceAxis()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1983 const bool memcpy_is_contiguous = slice->slice_strides(memcpy_dim) == 1; in HandleSlice()
2027 /*strides=*/slice->slice_strides(), /*builder=*/&b_); in HandleSlice()
|