Searched refs:valid_batch_size (Results 1 – 2 of 2) sorted by relevance
177 size_t valid_batch_size = 0; in xnn_setup_unpooling2d_nhwc_x32() local182 valid_batch_size = unpooling_op->valid_batch_size; in xnn_setup_unpooling2d_nhwc_x32()183 if (batch_size <= valid_batch_size) { in xnn_setup_unpooling2d_nhwc_x32()204 xnn_indirection_init_unpool2d(unpooling_op, valid_batch_size, 2 /* log2(sizeof(type32)) */); in xnn_setup_unpooling2d_nhwc_x32()232 unpooling_op->valid_batch_size = max(valid_batch_size, batch_size); in xnn_setup_unpooling2d_nhwc_x32()
243 size_t valid_batch_size; member