提交 cbd61c77 编写于 作者: W wanghaoshuang

fix crop function test

上级 701827f5
...@@ -34,8 +34,10 @@ TEST(Crop, real) { ...@@ -34,8 +34,10 @@ TEST(Crop, real) {
TensorShape outDims{numSamples, 2, 3, 3}; TensorShape outDims{numSamples, 2, 3, 3};
compare.addInputs( compare.addInputs(
BufferArg(VALUE_TYPE_FLOAT, test_grad ? outDims : inDims)); 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(); compare.run();
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册