| 32 | #include "tensorflow/core/platform/test_benchmark.h" |
| 33 | |
| 34 | GTEST_API_ int main(int argc, char** argv) { |
| 35 | std::cout << "Running main() from test_main.cc\n"; |
| 36 | |
| 37 | tensorflow::testing::InstallStacktraceHandler(); |
| 38 | testing::InitGoogleTest(&argc, argv); |
| 39 | for (int i = 1; i < argc; i++) { |
| 40 | if (absl::StartsWith(argv[i], "--benchmarks=")) { |
| 41 | const char* pattern = argv[i] + strlen("--benchmarks="); |
| 42 | tensorflow::testing::Benchmark::Run(pattern); |
| 43 | return 0; |
| 44 | } |
| 45 | } |
| 46 | return RUN_ALL_TESTS(); |
| 47 | } |
| 48 | #endif |
nothing calls this directly
no test coverage detected