Searched defs:split_dimension (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 42 int split_dimension; in __anon481f6f7f0102() local
|
| D | array_ops.cc | 574 DimensionHandle split_dimension; in __anonf6523ebd0a02() local
610 DimensionHandle split_dimension; in __anonf6523ebd0b02() local
|
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 93 def split(cls, tensor, split_dimension, num_devices): argument
221 split_dimension, argument
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 47 split_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 395 absl::optional<int64> split_dimension() const { return split_dimension_; } in split_dimension() function
|
| D | hlo_instruction.cc | 412 absl::optional<int64> split_dimension; in CreateFromProto() local
924 const absl::optional<int64>& split_dimension) { in CreateAllToAll()
|
| D | shape_inference.cc | 2018 const Shape& shape, int64 split_dimension, int64 concat_dimension, in InferAllToAllShape()
|
| D | hlo_parser.cc | 891 optional<int64> split_dimension; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 612 const absl::optional<int64>& split_dimension) in HloAllToAllInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1028 split_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2189 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64 split_dimension, in AllToAll()
3314 XlaOp AllToAll(const XlaOp operand, int64 split_dimension, in AllToAll()
|