提交 09f4f925 编写于 作者: W wangmeng28

Add unitest for factorization machine layer with sparse input

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