56 _fill_f = std::make_unique<NEFill>();
57 _fill_f->configure(output,
PixelValue(0,
input->info()->data_type(),
input->info()->quantization_info()));
59 _space_to_batch_kernel = std::make_unique<NESpaceToBatchLayerKernel>();
60 _space_to_batch_kernel->configure(
input, block_shape, paddings, output);
64 const int block_shape_x,
65 const int block_shape_y,
66 const Size2D &padding_left,
67 const Size2D &padding_right,
75 _fill_f = std::make_unique<NEFill>();
76 _fill_f->configure(output,
PixelValue(0,
input->info()->data_type(),
input->info()->quantization_info()));
78 _space_to_batch_kernel = std::make_unique<NESpaceToBatchLayerKernel>();
79 _space_to_batch_kernel->configure(
input, block_shape_x, block_shape_y, padding_left, padding_right, output);
93 const int block_shape_x,
94 const int block_shape_y,
95 const Size2D &padding_left,
96 const Size2D &padding_right,