提交 2ce8f187 编写于 作者: W wangmeng28

Fix tests for factorization machine layer

上级 22c5d1f1
...@@ -2373,7 +2373,6 @@ void testFactorizationMachineLayer(InputType type, bool useGpu) { ...@@ -2373,7 +2373,6 @@ void testFactorizationMachineLayer(InputType type, bool useGpu) {
TEST(Layer, FactorizationMachineLayer) { TEST(Layer, FactorizationMachineLayer) {
testFactorizationMachineLayer(INPUT_DATA, false); testFactorizationMachineLayer(INPUT_DATA, false);
testFactorizationMachineLayer(INPUT_DATA, true); testFactorizationMachineLayer(INPUT_DATA, true);
testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, false);
} }
int main(int argc, char** argv) { int main(int argc, char** argv) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册