Skip to content

Commit 5b3ff57

Browse files
fix_io_copy (#7729)
1 parent 767aeea commit 5b3ff57

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

lite/kernels/metal/image_op/io_copy_image_compute.mm

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,7 @@ void Run() override {
6262
auto input_dims = param.x->dims();
6363
auto output_dims = param.y->dims();
6464

65-
if ((input_dims.size() == 4 && input_dims[1] <= 4) ||
66-
(input_dims.size() == 3 && input_dims[0] <= 4)) {
65+
if (input_dims.size() == 4 || input_dims.size() == 3) {
6766
auto src = param.x->template data<float>();
6867
output_buffer_->CopyFromNCHW<float>(src);
6968
} else {

0 commit comments

Comments
 (0)