Home
last modified time | relevance | path

Searched refs:GradSource (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/_dynamo/variables/
Doptimizer.py15 GradSource,
264 grad_source = GradSource(
Dbuilder.py71 GradSource,
1604 if isinstance(source, GradSource) and is_from_optimizer_source(source):
/external/pytorch/torch/_dynamo/
Dsource.py219 class GradSource(ChainedSource): class
Dguards.py91 GradSource,
489 return source.is_dict_key() and not isinstance(source, GradSource)
903 elif istype(source, GradSource):