Searched refs:pytorch_qnnp_indirection_set_step_dimensions (Results 1 – 5 of 5) sorted by relevance
39 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_set_step_dimensions(
205 pytorch_qnnp_indirection_set_step_dimensions(max_pooling); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
266 pytorch_qnnp_indirection_set_step_dimensions(average_pooling); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
458 void pytorch_qnnp_indirection_set_step_dimensions(pytorch_qnnp_operator_t op) { in pytorch_qnnp_indirection_set_step_dimensions() function
786 pytorch_qnnp_indirection_set_step_dimensions(convolution); in pytorch_qnnp_setup_convolution_ndhwc_q8()