Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_tp_transform.py212 input_specs=_get_input_node_specs(node, placement_strategies),
239 else _get_input_node_specs(node, placement_strategies),
487 def _get_input_node_specs( function