Home
last modified time | relevance | path

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

12

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
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()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
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;
Dqhardsigmoid.cpp55 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_hardsigmoid()
Dqtanh.cpp58 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_tanh()
DChannelShuffle.cpp74 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in quantized_channel_shuffle_impl()
Dqhardswish.cpp53 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_hardswish()
Dqsigmoid.cpp63 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> in qnnpack_sigmoid()
Dqsoftmax.cpp99 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> softmax_op( in qsoftmax_qnnpack()
/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
Dqnnpack_utils.h33 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter>
Dqlinear_dynamic.cpp104 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter>( in apply_dynamic_impl()

12