MixedPrecisionPolicy (definition) in projects: aosp_15_r20

Searched defs:MixedPrecisionPolicy (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D	_fsdp_param.py	`219 mp_policy: MixedPrecisionPolicy, 370 def init_dtype_attrs(self, mp_policy: MixedPrecisionPolicy):`
H A D	_fsdp_api.py	`9 class MixedPrecisionPolicy: class`
H A D	fully_shard.py	`37 mp_policy: MixedPrecisionPolicy = MixedPrecisionPolicy(),`
H A D	_fsdp_state.py	`90 mp_policy: MixedPrecisionPolicy,`
H A D	_fsdp_param_group.py	`109 mp_policy: MixedPrecisionPolicy,`
/aosp_15_r20/external/pytorch/test/distributed/_tools/
H A D	test_fsdp2_mem_tracker.py	`77 mp_policy: MixedPrecisionPolicy,`
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A D	test_fully_shard_mixed_precision.py	`448 def _test_norm_modules(self, mp_policy: MixedPrecisionPolicy):`