Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dslice_test.cc393 std::array<int64_t, 2> slice_limits; member
478 std::array<int64_t, 4> slice_limits; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_to_all_decomposer.cc97 DimensionVector slice_limits(all_to_all->shape().dimensions().begin(), in ExpandInstruction() local
H A Dhlo_constant_folding_test.cc157 const int64_t slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
H A Dhlo_instructions.h1063 int64_t slice_limits(int64_t dimension) const { in slice_limits() function
1066 const std::vector<int64_t>& slice_limits() const { return slice_limits_; } in slice_limits() function
H A Dhlo_sharding_util.cc717 slice_limits(hlo.shape().rank()); in GatherEffectiveOutputSharding() local
846 std::vector<int64_t> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
H A Dhlo_instruction.cc376 std::vector<int64_t> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
4510 int64_t HloInstruction::slice_limits(int64_t dimension) const { in slice_limits() function in xla::HloInstruction
4514 const std::vector<int64_t>& HloInstruction::slice_limits() const { in slice_limits() function in xla::HloInstruction
H A Dalgebraic_simplifier.cc4914 std::vector<int64_t> slice_limits(rank); in HandleDynamicSlice() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.cc294 std::vector<int64_t> slice_limits( in HandleCustomCallSPMDInternal_RotateRight() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc229 auto slice_limits = llvm::to_vector<4>(type.getShape()); in SliceInMinorDims() local