Home
last modified time | relevance | path

Searched refs:max_num_shards (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/distributed/tensor/_ops/
D_tensor_ops.py94 if self_strategy.max_num_shards() >= other_strategy.max_num_shards()
D_pointwise_ops.py438 arg_max_shards = arg_strategy.max_num_shards()
/external/pytorch/torch/distributed/tensor/
D_op_schema.py142 def max_num_shards(self) -> int: member in OpStrategy