const ::arm_compute::Coordinates axises)
{
const int batches = input_shape[axises[0]];
- const int input_depth = input_shape[axises[3]];
- const int output_depth = output_shape[axises[3]];
const int input_height = input_shape[axises[1]];
const int input_width = input_shape[axises[2]];
- const int filter_batch = filter_shape[axises[0]];
+ const int input_depth = input_shape[axises[3]];
+
const int filter_height = filter_shape[axises[1]];
const int filter_width = filter_shape[axises[2]];
- const int filter_depth = filter_shape[axises[3]];
+
const int output_height = output_shape[axises[1]];
const int output_width = output_shape[axises[2]];
+ const int output_depth = output_shape[axises[3]];
const int pad_width = (stride_width * (input_width - 1) + filter_width - output_width) / 2;
const int pad_height = (stride_height * (input_height - 1) + filter_height - output_height) / 2;