Searched defs:param_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/distributed/_composable/fsdp/ |
| D | _fsdp_api.py | 44 param_dtype: Optional[torch.dtype] = None variable in MixedPrecisionPolicy
|
| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_mixed_precision.py | 165 def __init__(self, param_dtype, buffer_name="buffer", run_checks=True): argument
247 self, param_dtype, run_checks, *fsdp_args, **fsdp_kwargs argument
268 self, param_dtype, run_checks, *fsdp_args, **fsdp_kwargs argument
282 def _get_simple_model(self, param_dtype, *fsdp_args, **fsdp_kwargs): argument
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | api.py | 220 param_dtype: Optional[torch.dtype] = None variable in MixedPrecision
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 300 param_dtype=dtypes.float32): argument
|
| /external/pytorch/torch/nn/parallel/ |
| D | distributed.py | 89 param_dtype: Optional[torch.dtype] = None variable in _MixedPrecision
|