fix crop function test

86bdb2f3 · wanghaoshuang · 5e6e1f63 · 86bdb2f3
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

paddle/function/CropOpTest.cpp paddle/function/CropOpTest.cpp +4 -2

未找到文件。
--- a/paddle/function/CropOpTest.cpp
+++ b/paddle/function/CropOpTest.cpp
@@ -34,8 +34,10 @@ TEST(Crop, real) {
            TensorShape outDims{numSamples, 2, 3, 3};
            compare.addInputs(
                BufferArg(VALUE_TYPE_FLOAT, test_grad ? outDims : inDims));
-            compare.addOutputs(BufferArg(
+            compare.addOutputs(BufferArg(VALUE_TYPE_FLOAT,
-                VALUE_TYPE_FLOAT, test_grad ? inDims : outDims, ASSIGN_TO));
+                                         test_grad ? inDims : outDims,
+                                         tes_grad ? ADD_TO : ASSIGN_TO),
+                               test_grad ? ADD_TO : ASSIGN_TO);
            compare.run();
          }
        }