Home
last modified time | relevance | path

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

/external/pytorch/test/
Dtest_nnapi.py341 for mod_class in [ArgPlusConst, ConstPlusArg]:
343 with self.subTest(mod_class=mod_class.__name__, use_nhwc=use_nhwc):
348 self.check(mod_class(), arg, expected_memory_format=memory_format)
/external/pytorch/torch/
D_jit_internal.py1105 def _get_overloaded_methods(method, mod_class): argument
1113 overloads = class_name_map.get(mod_class.__name__, None)
1118 mod_class_fileno = get_source_lines_and_file(mod_class)[1]
1119 mod_end_fileno = mod_class_fileno + len(get_source_lines_and_file(mod_class)[0])
/external/pytorch/test/quantization/jit/
Dtest_quantize_jit.py2671 tracing, inplace, mod_class = test_case
2672 m = mod_class(inplace=inplace)