Searched defs:split_dimension (Results 1 – 14 of 14) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/ |
| D | IScheduler.h | 77 : _split_dimension(split_dimension), _strategy(strategy), _threshold(threshold) in _split_dimension() argument
86 Hints &set_split_dimension(unsigned int split_dimension) in set_split_dimension()
95 unsigned int split_dimension() const in split_dimension() function
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 42 int split_dimension; in __anona0f92a200102() local
|
| D | array_ops.cc | 574 DimensionHandle split_dimension; in __anon5ce23fde0a02() local
610 DimensionHandle split_dimension; in __anon5ce23fde0b02() local
|
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 129 def split(cls, tensor, split_dimension, num_devices, input_shape=None): argument
299 split_dimension, argument
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 46 split_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 50 const int split_dimension, in CreateSplitOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 520 absl::optional<int64> split_dimension() const { return split_dimension_; } in split_dimension() function
|
| D | hlo_instruction.cc | 483 absl::optional<int64> split_dimension; in CreateFromProto() local
1218 const absl::optional<int64>& split_dimension) { in CreateAllToAll()
|
| D | shape_inference.cc | 2143 const Shape& shape, int64_t split_dimension, int64_t concat_dimension, in InferAllToAllShape()
|
| D | hlo_parser.cc | 1365 optional<int64> split_dimension; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 782 const absl::optional<int64>& split_dimension) in HloAllToAllInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2946 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64_t split_dimension, in AllToAll()
2960 XlaOp XlaBuilder::AllToAllArray(XlaOp operand, int64_t split_dimension, in AllToAllArray()
3012 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64_t split_dimension, in AllToAllTuple()
4539 XlaOp AllToAll(const XlaOp operand, int64_t split_dimension, in AllToAll()
4547 XlaOp AllToAllTuple(const XlaOp operand, int64_t split_dimension, in AllToAllTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1696 int64_t channel_id, absl::optional<int64> split_dimension) { in GetPerGroupCollectiveOpsCreator()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 840 int split_dimension = sharding_it->first; in CreateOrGetSplitNodesForInputSharding() local
|