Home
last modified time | relevance | path

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

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