We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 767aeea commit 5b3ff57Copy full SHA for 5b3ff57
lite/kernels/metal/image_op/io_copy_image_compute.mm
@@ -62,8 +62,7 @@ void Run() override {
62
auto input_dims = param.x->dims();
63
auto output_dims = param.y->dims();
64
65
- if ((input_dims.size() == 4 && input_dims[1] <= 4) ||
66
- (input_dims.size() == 3 && input_dims[0] <= 4)) {
+ if (input_dims.size() == 4 || input_dims.size() == 3) {
67
auto src = param.x->template data<float>();
68
output_buffer_->CopyFromNCHW<float>(src);
69
} else {
0 commit comments