Home
last modified time | relevance | path

Searched refs:pytorch_qnnp_operator (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Doperator.h57 struct pytorch_qnnp_operator { struct
140 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_output_element_size() argument
145 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_input_element_size()
150 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_kernel_element_size()
155 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_bias_element_size()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dchannel-shuffle.c54 channel_shuffle_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_channel_shuffle_nc_x8()
58 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_channel_shuffle_nc_x8()
H A Dclamp.c54 clamp_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_clamp_nc_u8()
58 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_clamp_nc_u8()
H A Dsoftargmax.c77 softargmax_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_softargmax_nc_q8()
81 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_softargmax_nc_q8()
H A Dadd.c101 add_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_add_nc_q8()
105 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_add_nc_q8()
H A Dglobal-average-pooling.c76 global_average_pooling_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_global_average_pooling_nwc_q8()
80 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_global_average_pooling_nwc_q8()
H A Dsigmoid.c89 sigmoid_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_sigmoid_nc_q8()
93 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_sigmoid_nc_q8()
H A Dhardswish.c72 hardswish_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_hardswish_nc_q8()
76 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_hardswish_nc_q8()
H A Dtanh.c89 tanh_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_tanh_nc_q8()
93 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_tanh_nc_q8()
H A Dhardsigmoid.c89 hardsigmoid_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_hardsigmoid_nc_q8()
93 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_hardsigmoid_nc_q8()
H A Dleaky-relu.c98 leaky_relu_op = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_leaky_relu_nc_q8()
102 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_leaky_relu_nc_q8()
H A Dfully-connected.c60 fully_connected = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_fully_connected_nc_q8()
64 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_fully_connected_nc_q8()
H A Dfully-connected-sparse.c65 fully_connected = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
69 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
H A Dmax-pooling.c110 max_pooling = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
114 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
H A Daverage-pooling.c138 average_pooling = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
142 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
H A Ddeconvolution.c117 deconvolution = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
121 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
H A Dconvolution.c188 convolution = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_convolution_ndhwc_q8()
192 sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_convolution_ndhwc_q8()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DQnnpackUtils.h203 calloc(1, sizeof(struct pytorch_qnnp_operator))); in PackedConvWeightsQnnp()
207 sizeof(struct pytorch_qnnp_operator)); in PackedConvWeightsQnnp()
211 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter>( in PackedConvWeightsQnnp()
291 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> convolution_op;
H A Dqhardsigmoid.cpp55 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_hardsigmoid()
H A Dqtanh.cpp58 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_tanh()
H A DChannelShuffle.cpp74 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in quantized_channel_shuffle_impl()
H A Dqhardswish.cpp53 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_hardswish()
H A Dqsigmoid.cpp63 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_sigmoid()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqnnpack_utils.h33 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter>
H A Dqlinear_dynamic.cpp104 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter>( in apply_dynamic_impl()

12