Searched defs:split_dimension (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 92 def split(cls, tensor, split_dimension, num_devices): argument
184 def split(tensor, split_dimension, num_devices): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 42 int split_dimension; in __anonc4e6eef40102() local
|
| D | array_ops.cc | 568 DimensionHandle split_dimension; in __anondb9326b20a02() local
604 DimensionHandle split_dimension; in __anondb9326b20b02() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 46 split_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 499 const LocalOp& operand, int64 split_dimension, int64 concat_dimension, in AllToAll()
|
| D | xla_client.py | 1207 split_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1992 const Shape& shape, int64 split_dimension, int64 concat_dimension, in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2159 XlaOp XlaBuilder::AllToAll(const XlaOp& operand, int64 split_dimension, in AllToAll()
3244 XlaOp AllToAll(const XlaOp& operand, int64 split_dimension, in AllToAll()
|