Searched defs:zero_point_base (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_workflow_ops.py | 404 def _test_learnable_forward_per_tensor(self, X, device, scale_base, zero_point_base): argument
449 def _test_learnable_backward_per_tensor(self, X, device, scale_base, zero_point_base): argument
784 … def _test_learnable_forward_per_channel(self, X_base, device, scale_base, zero_point_base, axis): argument
893 … def _test_learnable_backward_per_channel(self, X_base, device, scale_base, zero_point_base, axis): argument
|