Searched refs:max_pooling (Results 1 – 4 of 4) sorted by relevance
51 pytorch_qnnp_operator_t max_pooling = NULL; in pytorch_qnnp_create_max_pooling2d_nhwc_u8() local110 max_pooling = calloc(1, sizeof(struct pytorch_qnnp_operator)); in pytorch_qnnp_create_max_pooling2d_nhwc_u8()111 if (max_pooling == NULL) { in pytorch_qnnp_create_max_pooling2d_nhwc_u8()118 max_pooling->input_padding_height = input_padding_height; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()119 max_pooling->input_padding_width = input_padding_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()121 max_pooling->kernel_height = pooling_height; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()122 max_pooling->kernel_width = pooling_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()123 max_pooling->stride_height = stride_height; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()124 max_pooling->stride_width = stride_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()125 max_pooling->dilation_height = dilation_height; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()[all …]
270 pytorch_qnnp_operator_t* max_pooling);273 pytorch_qnnp_operator_t max_pooling,
311 max_pooling_op->context.max_pooling = (struct max_pooling_context) { in setup_max_pooling2d_nhwc()326 memcpy(&max_pooling_op->context.max_pooling.params, params, params_size); in setup_max_pooling2d_nhwc()
284 struct max_pooling_context max_pooling; member