| D | _quant_ops.py | 183 def infer_dtype(self, x_type, min_type, max_type): argument
247 def infer_dtype(self, x_type, min_type, max_type): argument
621 def infer_dtype(self, x_type, min_type, max_type): argument
686 def infer_dtype(self, dout_type, x_type, min_type, max_type): argument
737 def infer_dtype(self, x_type, min_type, max_type): argument
796 def infer_dtype(self, dout_type, x_type, min_type, max_type): argument
804 def _fake_quant_per_infer_dtype(prim_name, x_type, min_type, max_type): argument
815 def _fake_quant_per_grad_infer_dtype(prim_name, dout_type, x_type, min_type, max_type): argument
894 def infer_dtype(self, x_type, min_type, max_type): argument
942 def infer_dtype(self, dout_type, x_type, min_type, max_type): argument
[all …]
|