Searched defs:single_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 450 single_dim=-1, argument
479 single_dim=-1, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 1049 int32 single_dim; in __anonb720852f0f02() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 5017 OpSharding GetManualSharding(const OpSharding& original, int64_t single_dim) { in GetManualSharding()
5061 XlaBuilder* builder, XlaOp input, int single_dim, in ConvertSpmdFullToShardShape()
5106 XlaBuilder* builder, XlaOp input, const Shape& output_shape, int single_dim, in ConvertSpmdShardToFullShape()
|