Searched defs:slice_strides (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 96 DimensionVector slice_strides(all_to_all->shape().rank(), 1); in ExpandInstruction() local
|
| D | hlo_constant_folding_test.cc | 157 const int64 slice_strides[] = {1, 1, 1, 1, 1}; in TEST_F() local
|
| D | hlo_instructions.h | 866 int64 slice_strides(int64_t dimension) const { in slice_strides() function
869 const std::vector<int64>& slice_strides() const { return slice_strides_; } in slice_strides() function
|
| D | hlo_instruction.cc | 312 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
4251 int64 HloInstruction::slice_strides(int64_t dimension) const { in slice_strides() function in xla::HloInstruction
4255 const std::vector<int64>& HloInstruction::slice_strides() const { in slice_strides() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 4545 std::vector<int64> slice_strides(rank, 1); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 392 std::array<int64, 2> slice_strides; member
477 std::array<int64, 4> slice_strides; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 289 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1808 DenseIntElementsAttr slice_strides = in matchAndRewrite() local
1925 DenseIntElementsAttr slice_strides = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 881 std::vector<int64> slice_strides(self_shape.rank(), 1); in ExchangeHalo() local
|
| D | spmd_partitioner.cc | 3012 std::vector<int64> slice_strides(slice_shape.rank(), 1); in HandleOutfeed() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3924 NodeDef* slice_strides; in GetStridedSliceAxis() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 278 llvm::SmallVector<int64_t, 4> slice_strides(type.getRank(), 1); in SliceInMinorDims() local
|