Home
last modified time | relevance | path

Searched refs:pytorch_qnnp_indirection_set_step_dimensions (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Dindirection.h39 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_set_step_dimensions(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dmax-pooling.c205 pytorch_qnnp_indirection_set_step_dimensions(max_pooling); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
H A Daverage-pooling.c266 pytorch_qnnp_indirection_set_step_dimensions(average_pooling); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
H A Dindirection.c458 void pytorch_qnnp_indirection_set_step_dimensions(pytorch_qnnp_operator_t op) { in pytorch_qnnp_indirection_set_step_dimensions() function
H A Dconvolution.c786 pytorch_qnnp_indirection_set_step_dimensions(convolution); in pytorch_qnnp_setup_convolution_ndhwc_q8()