Home
last modified time | relevance | path

Searched refs:NNModuleToString (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/repro/
H A Dafter_dynamo.py26 NNModuleToString,
165 model_str = NNModuleToString.convert(gm)
242 assert NNModuleToString.can_convert_to_string(gm)
H A Dafter_aot.py33 NNModuleToString,
235 model_str += NNModuleToString.convert(gm)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Ddebug_utils.py128 class NNModuleToString: class
157 if type(module) not in NNModuleToString.safe_reprs: