Searched defs:MixedPrecision (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_unshard_params.py | 122 mixed_precision: Optional[MixedPrecision],
284 mixed_precision: Optional[MixedPrecision],
371 mixed_precision: Optional[MixedPrecision],
|
| H A D | test_fsdp_pure_fp16.py | 99 mixed_precision: MixedPrecision,
|
| H A D | test_fsdp_fine_tune.py | 339 mixed_precision: MixedPrecision,
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/ |
| H A D | fully_shard.py | 54 mixed_precision: Optional[MixedPrecision] = None,
|
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | api.py | 112 class MixedPrecision: class
|
| H A D | _init_utils.py | 432 mixed_precision: Optional[MixedPrecision],
|
| H A D | fully_sharded_data_parallel.py | 421 mixed_precision: Optional[MixedPrecision] = None,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 1205 mixed_precision: Optional[MixedPrecision] = None,
1295 mixed_precision: Optional[MixedPrecision] = None,
|