Home
last modified time | relevance | path

Searched refs:pytorch_qnnp_operator_t (Results 1 – 25 of 66) sorted by relevance

123

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/
Dpytorch_qnnpack.h46 typedef struct pytorch_qnnp_operator* pytorch_qnnp_operator_t; typedef
70 pytorch_qnnp_operator_t* convolution);
98 pytorch_qnnp_operator_t* convolution);
101 pytorch_qnnp_operator_t convolution,
112 pytorch_qnnp_operator_t convolution,
146 pytorch_qnnp_operator_t* deconvolution);
149 pytorch_qnnp_operator_t deconvolution,
171 pytorch_qnnp_operator_t* fully_connected);
190 pytorch_qnnp_operator_t* fully_connected);
193 pytorch_qnnp_operator_t fully_connected,
[all …]
Dqnnpack_func.h10 const pytorch_qnnp_operator_t convolution,
119 const pytorch_qnnp_operator_t convolution,
136 const pytorch_qnnp_operator_t deconvolution,
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
Dindirection.h22 pytorch_qnnp_operator_t op,
27 pytorch_qnnp_operator_t op,
31 pytorch_qnnp_operator_t op,
36 pytorch_qnnp_operator_t op,
40 pytorch_qnnp_operator_t op);
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dchannel-shuffle.c24 pytorch_qnnp_operator_t* channel_shuffle_out) { in pytorch_qnnp_create_channel_shuffle_nc_x8()
25 pytorch_qnnp_operator_t channel_shuffle_op = NULL; in pytorch_qnnp_create_channel_shuffle_nc_x8()
77 pytorch_qnnp_operator_t channel_shuffle_op, in pytorch_qnnp_setup_channel_shuffle_nc_x8()
Dclamp.c24 pytorch_qnnp_operator_t* clamp_out) { in pytorch_qnnp_create_clamp_nc_u8()
25 pytorch_qnnp_operator_t clamp_op = NULL; in pytorch_qnnp_create_clamp_nc_u8()
78 pytorch_qnnp_operator_t clamp, in pytorch_qnnp_setup_clamp_nc_u8()
Dsoftargmax.c25 pytorch_qnnp_operator_t* softargmax_out) { in pytorch_qnnp_create_softargmax_nc_q8()
26 pytorch_qnnp_operator_t softargmax_op = NULL; in pytorch_qnnp_create_softargmax_nc_q8()
115 pytorch_qnnp_operator_t softargmax, in pytorch_qnnp_setup_softargmax_nc_q8()
Dadd.c32 pytorch_qnnp_operator_t* add_out) { in pytorch_qnnp_create_add_nc_q8()
33 pytorch_qnnp_operator_t add_op = NULL; in pytorch_qnnp_create_add_nc_q8()
132 pytorch_qnnp_operator_t add_op, in pytorch_qnnp_setup_add_nc_q8()
Dtanh.c28 pytorch_qnnp_operator_t* tanh_out) { in pytorch_qnnp_create_tanh_nc_q8()
29 pytorch_qnnp_operator_t tanh_op = NULL; in pytorch_qnnp_create_tanh_nc_q8()
137 pytorch_qnnp_operator_t tanh, in pytorch_qnnp_setup_tanh_nc_q8()
Dsigmoid.c28 pytorch_qnnp_operator_t* sigmoid_out) { in pytorch_qnnp_create_sigmoid_nc_q8()
29 pytorch_qnnp_operator_t sigmoid_op = NULL; in pytorch_qnnp_create_sigmoid_nc_q8()
135 pytorch_qnnp_operator_t sigmoid, in pytorch_qnnp_setup_sigmoid_nc_q8()
Dglobal-average-pooling.c30 pytorch_qnnp_operator_t* global_average_pooling_out) { in pytorch_qnnp_create_global_average_pooling_nwc_q8()
31 pytorch_qnnp_operator_t global_average_pooling_op = NULL; in pytorch_qnnp_create_global_average_pooling_nwc_q8()
115 pytorch_qnnp_operator_t global_average_pooling_op, in pytorch_qnnp_setup_global_average_pooling_nwc_q8()
Dhardswish.c28 pytorch_qnnp_operator_t* hardswish_out) { in pytorch_qnnp_create_hardswish_nc_q8()
29 pytorch_qnnp_operator_t hardswish_op = NULL; in pytorch_qnnp_create_hardswish_nc_q8()
123 pytorch_qnnp_operator_t hardswish, in pytorch_qnnp_setup_hardswish_nc_q8()
Dleaky-relu.c29 pytorch_qnnp_operator_t* leaky_relu_out) { in pytorch_qnnp_create_leaky_relu_nc_q8()
30 pytorch_qnnp_operator_t leaky_relu_op = NULL; in pytorch_qnnp_create_leaky_relu_nc_q8()
145 pytorch_qnnp_operator_t leaky_relu, in pytorch_qnnp_setup_leaky_relu_nc_q8()
Dhardsigmoid.c28 pytorch_qnnp_operator_t* hardsigmoid_out) { in pytorch_qnnp_create_hardsigmoid_nc_q8()
29 pytorch_qnnp_operator_t hardsigmoid_op = NULL; in pytorch_qnnp_create_hardsigmoid_nc_q8()
140 pytorch_qnnp_operator_t hardsigmoid, in pytorch_qnnp_setup_hardsigmoid_nc_q8()
Dfully-connected.c36 pytorch_qnnp_operator_t* fully_connected_out) { in pytorch_qnnp_create_fully_connected_nc_q8()
37 pytorch_qnnp_operator_t fully_connected = NULL; in pytorch_qnnp_create_fully_connected_nc_q8()
131 pytorch_qnnp_operator_t fully_connected, in pytorch_qnnp_setup_fully_connected_nc_q8()
Dfully-connected-sparse.c41 pytorch_qnnp_operator_t* fully_connected_out) { in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
42 pytorch_qnnp_operator_t fully_connected = NULL; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
139 pytorch_qnnp_operator_t fully_connected, in pytorch_qnnp_setup_fully_connected_sparse_dq_nc_q8()
Dmax-pooling.c50 pytorch_qnnp_operator_t* max_pooling_out) { in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
51 pytorch_qnnp_operator_t max_pooling = NULL; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
144 pytorch_qnnp_operator_t max_pooling, in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
Daverage-pooling.c47 pytorch_qnnp_operator_t* average_pooling_out) { in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
48 pytorch_qnnp_operator_t average_pooling = NULL; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
205 pytorch_qnnp_operator_t average_pooling, in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
Dconvolution.c69 pytorch_qnnp_operator_t* convolution_out, in pytorch_qnnp_create_convolution_ndhwc_q8()
71 pytorch_qnnp_operator_t convolution = NULL; in pytorch_qnnp_create_convolution_ndhwc_q8()
571 pytorch_qnnp_operator_t* convolution_out) { in pytorch_qnnp_create_convolution2d_nhwc_q8()
628 pytorch_qnnp_operator_t* convolution_out) { in pytorch_qnnp_create_convolution3d_ndhwc_q8()
660 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution2d_nhwc_q8()
683 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution_ndhwc_q8()
Ddeconvolution.c61 pytorch_qnnp_operator_t* deconvolution_out) { in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
62 pytorch_qnnp_operator_t deconvolution = NULL; in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
221 pytorch_qnnp_operator_t deconvolution, in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
Dindirection.c18 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_conv3d()
225 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_dwconv()
340 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_deconv2d()
409 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_maxpool2d()
458 void pytorch_qnnp_indirection_set_step_dimensions(pytorch_qnnp_operator_t op) { in pytorch_qnnp_indirection_set_step_dimensions()
Doperator-delete.c15 pytorch_qnnp_operator_t op) { in pytorch_qnnp_delete_operator()
Ddeconv-run.cc84 void operator()(pytorch_qnnp_operator_t op) { in operator ()()
90 const pytorch_qnnp_operator_t deconvolution, in qnnpackDeConv()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
Dadd.cc38 pytorch_qnnp_operator_t addOperator = nullptr; in add_nc_q8()
106 pytorch_qnnp_operator_t addOperator = nullptr; in add_nc_q8_inplace()
Dglobal-average-pooling.cc45 pytorch_qnnp_operator_t globalPoolingOperator = nullptr; in global_average_pooling_q8()
Dsoftargmax.cc37 pytorch_qnnp_operator_t softArgMaxOperator = nullptr; in softargmax_q8()

123