Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/_tensor/experimental/
H A Dtest_tp_transform.py7 tensor_parallel_transformation,
79 tp_exported_program = tensor_parallel_transformation(
117 tp_exported_program = tensor_parallel_transformation(
153 tp_exported_program = tensor_parallel_transformation(
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_tp_transform.py34 def tensor_parallel_transformation( function