Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dall_to_all_decomposer.cc97 DimensionVector slice_limits(all_to_all->shape().dimensions().begin(), in ExpandInstruction() local
Dhlo_constant_folding_test.cc156 const int64 slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
Dhlo_instructions.h859 int64 slice_limits(int64_t dimension) const { in slice_limits() function
862 const std::vector<int64>& slice_limits() const { return slice_limits_; } in slice_limits() function
Dhlo_sharding_util.cc655 slice_limits(hlo.shape().rank()); in GatherEffectiveOutputSharding() local
766 std::vector<int64> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
Dhlo_instruction.cc312 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
4239 int64 HloInstruction::slice_limits(int64_t dimension) const { in slice_limits() function in xla::HloInstruction
4243 const std::vector<int64>& HloInstruction::slice_limits() const { in slice_limits() function in xla::HloInstruction
Dalgebraic_simplifier.cc4544 std::vector<int64> slice_limits(rank); in HandleDynamicSlice() local
/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/spmd/
Dcustom_call_handler.cc292 std::vector<int64> slice_limits( in HandleCustomCallSPMDInternal_RotateRight() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc126 SmallVector<int64_t, 4> slice_limits; in BuildSliceLimits() local
1806 DenseIntElementsAttr slice_limits = BuildSliceLimits( in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc275 auto slice_limits = llvm::to_vector<4>(type.getShape()); in SliceInMinorDims() local