is_bw (definition) in projects: external

Searched defs:is_bw (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/utils/
D	module_tracker.py	`78 def is_bw(self): member in ModuleTracker 95 def _get_append_fn(self, name, is_bw): argument 109 def _get_pop_fn(self, name, is_bw): argument`
/external/pytorch/torch/distributed/_tools/
D	mod_tracker.py	`80 def is_bw(self): member in ModTracker 165 def _get_append_fn(self, w_mod, name, is_bw): argument 190 def _get_pop_fn(self, w_mod, name, is_bw): argument`