Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_parallelize_api.py11 PrepareModuleOutput,
192 PrepareModuleOutput(
H A Dtest_tp_style.py20 PrepareModuleOutput,
337 prepare_out_style = PrepareModuleOutput(
/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A D__init__.py8 PrepareModuleOutput,
H A Dstyle.py463 class PrepareModuleOutput(ParallelStyle): class
/aosp_15_r20/external/pytorch/docs/source/
H A Ddistributed.tensor.parallel.rst45 .. autoclass:: torch.distributed.tensor.parallel.PrepareModuleOutput
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_dtensor_compile.py34 PrepareModuleOutput,
755 prepare_module_out = PrepareModuleOutput(