diff --git a/onnxruntime/test/onnx/main.cc b/onnxruntime/test/onnx/main.cc index 7d31f6e377..80d67ff166 100644 --- a/onnxruntime/test/onnx/main.cc +++ b/onnxruntime/test/onnx/main.cc @@ -344,18 +344,12 @@ int real_main(int argc, char* argv[], Ort::Env& env) { {"convtranspose_3d", "disable reason"}, {"cast_STRING_to_FLOAT", "Cast opset 9 not supported yet"}, {"cast_FLOAT_to_STRING", "Cast opset 9 not supported yet"}, - {"tf_inception_resnet_v2", "Cast opset 9 not supported yet"}, - {"tf_inception_v4", "Cast opset 9 not supported yet"}, {"tf_nasnet_large", "disable temporarily"}, {"tf_nasnet_mobile", "disable temporarily"}, {"tf_pnasnet_large", "disable temporarily"}, {"shrink", "test case is wrong", {"onnx141"}}, {"maxpool_with_argmax_2d_precomputed_strides", "ShapeInferenceError"}, {"tf_inception_v2", "result mismatch"}, - {"tf_mobilenet_v2_1.0_224", "result mismatch"}, - {"tf_mobilenet_v2_1.4_224", "result mismatch"}, - {"tf_mobilenet_v1_1.0_224", "result mismatch"}, - {"mobilenetv2-1.0", "result mismatch"}, {"mxnet_arcface", "result mismatch"} }; @@ -365,6 +359,13 @@ int real_main(int argc, char* argv[], Ort::Env& env) { broken_tests.insert({"quantizelinear", "ambiguity in scalar dimensions [] vs [1]", {"onnx150"}}); #endif +#ifdef USE_MKLDNN + broken_tests.insert({"tf_mobilenet_v2_1.0_224", "result mismatch"}); + broken_tests.insert({"tf_mobilenet_v2_1.4_224", "result mismatch"}); + broken_tests.insert({"tf_mobilenet_v1_1.0_224", "result mismatch"}); + broken_tests.insert({"mobilenetv2-1.0", "result mismatch"}); +#endif + #ifdef USE_OPENVINO broken_tests.insert({"fp16_shufflenet", "accuracy mismatch with fp16 precision"}); broken_tests.insert({"fp16_inception_v1", "accuracy mismatch with fp16 precision"});