Searched defs:slice_limits (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 97 DimensionVector slice_limits(all_to_all->shape().dimensions().begin(), in ExpandInstruction() local
|
| D | hlo_constant_folding_test.cc | 157 const int64_t slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
|
| D | hlo_instructions.h | 1063 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
|
| D | hlo_sharding_util.cc | 717 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
|
| D | hlo_instruction.cc | 376 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
|
| D | algebraic_simplifier.cc | 4914 std::vector<int64_t> slice_limits(rank); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 393 std::array<int64_t, 2> slice_limits; member
478 std::array<int64_t, 4> slice_limits; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 294 std::vector<int64_t> slice_limits( in HandleCustomCallSPMDInternal_RotateRight() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 229 auto slice_limits = llvm::to_vector<4>(type.getShape()); in SliceInMinorDims() local
|