Searched refs:_test_op_graph (Results 1 – 1 of 1) sorted by relevance
/external/pytorch/test/quantization/bc/ |
D | test_backward_compatibility.py | 125 def _test_op_graph( member in TestSerialization 361 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False) 379 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False) 399 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False) 419 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False) 439 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False) 459 self._test_op_graph(module, input_size=[1, 3, 6, 6], generate=False)
|