diff options
Diffstat (limited to 'infra/cmake/packages/TensorFlowVersionChecker.c')
-rw-r--r-- | infra/cmake/packages/TensorFlowVersionChecker.c | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/infra/cmake/packages/TensorFlowVersionChecker.c b/infra/cmake/packages/TensorFlowVersionChecker.c new file mode 100644 index 000000000..6161ef74a --- /dev/null +++ b/infra/cmake/packages/TensorFlowVersionChecker.c @@ -0,0 +1,9 @@ +#include <string.h> +#include <tensorflow/c/c_api.h> + +int main(int argc, char **argv) +{ + if (argc >= 2 && !strcmp(argv[1], TF_Version())) + return 0; + return 255; +} |