Searched defs:slice_strides (Results 1 – 13 of 13) 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 | 158 const int64_t slice_strides[] = {1, 1, 1, 1, 1}; in TEST_F() local
|
| D | hlo_instructions.h | 1070 int64_t slice_strides(int64_t dimension) const { in slice_strides() function
1073 const std::vector<int64_t>& slice_strides() const { return slice_strides_; } in slice_strides() function
|
| D | hlo_instruction.cc | 376 std::vector<int64_t> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
4522 int64_t HloInstruction::slice_strides(int64_t dimension) const { in slice_strides() function in xla::HloInstruction
4526 const std::vector<int64_t>& HloInstruction::slice_strides() const { in slice_strides() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 4915 std::vector<int64_t> slice_strides(rank, 1); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_util_ops.cc | 190 const std::vector<int64_t> slice_strides(rank, 1); in CompileInternal() local
338 const std::vector<int64_t> slice_strides(rank, 1); in CompileInternal() local
|
| D | strided_slice_op.cc | 290 absl::InlinedVector<int64_t, 4> slice_begin, slice_end, slice_strides; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 394 std::array<int64_t, 2> slice_strides; member
479 std::array<int64_t, 4> slice_strides; member
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 141 std::vector<int64_t> slice_strides(rank, 1); in AggregateToTopKBuilder() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 797 std::vector<int64_t> slice_strides(self_shape.rank(), 1); in ExchangeHalo() local
|
| D | spmd_partitioner.cc | 3797 std::vector<int64_t> 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 | 232 llvm::SmallVector<int64_t, 4> slice_strides(type.getRank(), 1); in SliceInMinorDims() local
|