Home
last modified time | relevance | path

Searched defs:slice_strides (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc381 std::array<int64, 2> slice_strides; member
466 std::array<int64, 4> slice_strides; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h627 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
Dhlo_constant_folding_test.cc155 const int64 slice_strides[] = {1, 1, 1, 1, 1}; in TEST_F() local
Dconvolution_group_converter.cc476 std::vector<int64> slice_strides(filter->shape().dimensions_size(), 1); in HandleConvolution() local
Dhlo_instruction.cc289 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
3263 int64 HloInstruction::slice_strides(int64 dimension) const { in slice_strides() function in xla::HloInstruction
3267 const std::vector<int64>& HloInstruction::slice_strides() const { in slice_strides() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc76 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3156 NodeDef* slice_strides; in GetSliceAxis() local