Home
last modified time | relevance | path

Searched defs:slice_limits (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc391 std::array<int64, 2> slice_limits; member
476 std::array<int64, 4> slice_limits; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h691 int64 slice_limits(int64 dimension) const { return slice_limits_[dimension]; } in slice_limits() function
692 const std::vector<int64>& slice_limits() const { return slice_limits_; } in slice_limits() function
Dhlo_constant_folding_test.cc154 const int64 slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
Dhlo_instruction.cc290 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
3588 int64 HloInstruction::slice_limits(int64 dimension) const { in slice_limits() function in xla::HloInstruction
3592 const std::vector<int64>& HloInstruction::slice_limits() const { in slice_limits() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dhlo_ops.cc92 SmallVector<int64_t, 4> slice_limits; in BuildSliceLimits() local