Searched defs:slice_limits (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 96 DimensionVector slice_limits(all_to_all->shape().dimensions().begin(), in ExpandInstruction() local
|
| D | hlo_constant_folding_test.cc | 156 const int64 slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
|
| D | hlo_instructions.h | 802 int64 slice_limits(int64 dimension) const { return slice_limits_[dimension]; } in slice_limits() function
803 const std::vector<int64>& slice_limits() const { return slice_limits_; } in slice_limits() function
|
| D | hlo_sharding_util.cc | 626 slice_limits(hlo.shape().rank()); in GatherEffectiveOutputSharding() local
737 std::vector<int64> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
|
| D | hlo_instruction.cc | 311 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
3942 int64 HloInstruction::slice_limits(int64 dimension) const { in slice_limits() function in xla::HloInstruction
3946 const std::vector<int64>& HloInstruction::slice_limits() const { in slice_limits() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 4398 std::vector<int64> slice_limits(rank); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 391 std::array<int64, 2> slice_limits; member
476 std::array<int64, 4> slice_limits; member
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 123 SmallVector<int64_t, 4> slice_limits; in BuildSliceLimits() local
1437 DenseIntElementsAttr slice_limits = BuildSliceLimits( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 298 auto slice_limits = llvm::to_vector<4>(type.getShape()); in SliceInMinorDims() local
|