Searched refs:node_info_to_non_tensor_args (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 3910 def _check_not_observed(self, model, node_info_to_non_tensor_args): argument
3925 indices = node_info_to_non_tensor_args.get(
3935 def _test_dtype_propagation(self, model, node_info_to_non_tensor_args, *args): argument
3939 self._check_not_observed(prepared_model, node_info_to_non_tensor_args)
3948 node_info_to_non_tensor_args = {NodeInfo("call_method", "masked_fill"): [1, 2]}
3949 self._test_dtype_propagation(model, node_info_to_non_tensor_args, *args)
3957 node_info_to_non_tensor_args = {NodeInfo("call_method", "permute"): [1, 2]}
3958 self._test_dtype_propagation(model, node_info_to_non_tensor_args, *args)
3966 node_info_to_non_tensor_args = {NodeInfo("call_method", "repeat"): [1, 2]}
3967 self._test_dtype_propagation(model, node_info_to_non_tensor_args, *args)
[all …]
|