Searched refs:valid_batch_size (Results 1 – 5 of 5) sorted by relevance
191 size_t valid_batch_size = 0; in pytorch_qnnp_setup_max_pooling2d_nhwc_u8() local195 valid_batch_size = max_pooling->valid_batch_size; in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()196 if (batch_size <= valid_batch_size) { in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()220 pytorch_qnnp_indirection_init_maxpool2d(max_pooling, valid_batch_size); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()225 max_pooling->valid_batch_size = max(valid_batch_size, batch_size); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
252 size_t valid_batch_size = 0; in pytorch_qnnp_setup_average_pooling2d_nhwc_q8() local256 valid_batch_size = average_pooling->valid_batch_size; in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()257 if (batch_size <= valid_batch_size) { in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()282 pytorch_qnnp_indirection_init_dwconv(average_pooling, valid_batch_size); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()287 average_pooling->valid_batch_size = max(valid_batch_size, batch_size); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
170 size_t valid_batch_size = 0; in xnn_setup_unpooling2d_nhwc_x32() local175 valid_batch_size = unpooling_op->valid_batch_size; in xnn_setup_unpooling2d_nhwc_x32()176 if (batch_size <= valid_batch_size) { in xnn_setup_unpooling2d_nhwc_x32()197 xnn_indirection_init_unpool2d(unpooling_op, valid_batch_size, 2 /* log2(sizeof(type32)) */); in xnn_setup_unpooling2d_nhwc_x32()225 unpooling_op->valid_batch_size = max(valid_batch_size, batch_size); in xnn_setup_unpooling2d_nhwc_x32()
110 size_t valid_batch_size; member
172 size_t valid_batch_size; member