Home
last modified time | relevance | path

Searched defs:op_scale (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset10.py808 g: jit_utils.GraphContext, q_input, q_weight, bias, op_scale, op_zero_point argument
822 g: jit_utils.GraphContext, q_input, q_weight, bias, op_scale, op_zero_point argument
836 def quantized_add(g: jit_utils.GraphContext, x, y, op_scale, op_zero_point): argument
846 def quantized_add_relu(g: jit_utils.GraphContext, x, y, op_scale, op_zero_point): argument
857 def quantized_mul(g: jit_utils.GraphContext, x, y, op_scale, op_zero_point): argument
867 def quantized_hardswish(g: jit_utils.GraphContext, x, op_scale, op_zero_point): argument
876 def quantized_sigmoid(g: jit_utils.GraphContext, x, op_scale, op_zero_point): argument
886 g: jit_utils.GraphContext, x, negative_slope, inplace, op_scale, op_zero_point argument
903 op_scale, argument
921 op_scale, argument
[all …]
H A Dsymbolic_opset13.py848 g: jit_utils.GraphContext, q_input, q_weight, bias, op_scale, op_zero_point argument
864 g: jit_utils.GraphContext, q_input, q_weight, bias, op_scale, op_zero_point argument
889 op_scale, argument
915 op_scale, argument
941 op_scale, argument
967 op_scale, argument
992 op_scale, argument
1017 op_scale, argument
1043 op_scale, argument
1071 op_scale, argument
[all …]
H A Dsymbolic_opset17.py81 op_scale, argument
H A Dsymbolic_opset14.py123 def quantized_hardswish(g: jit_utils.GraphContext, x, op_scale, op_zero_point): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_matmul_ops_common.h244 float op_scale = post_op_param.param[0]; in Setup() local
251 float op_scale = post_op_param.param[0]; in Setup() local
259 float op_scale = post_op_param.param[0]; in Setup() local
266 float op_scale = post_op_param.param[0]; in Setup() local
273 float op_scale = post_op_param.param[0]; in Setup() local
280 float op_scale = post_op_param.param[0]; in Setup() local
287 float op_scale = post_op_param.param[0]; in Setup() local
299 float op_scale = post_op_param.param[0]; in Setup() local
H A Dmkl_conv_ops.cc337 float op_scale = post_op_param.param[0]; in Setup() local
344 float op_scale = post_op_param.param[0]; in Setup() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DXnnpackUtils.h63 float op_scale, in xnnp_create_convolution2d_nhwc()