56 _batch_to_space_kernel->configure(compile_context,
input, block_shape, output);
67 int32_t block_shape_x,
68 int32_t block_shape_y,
73 _batch_to_space_kernel->configure(compile_context,
input, block_shape_x, block_shape_y, output, crop_info);
83 int32_t block_shape_x,
84 int32_t block_shape_y,