diff --git a/llvm/cmake/modules/TensorFlowCompile.cmake b/llvm/cmake/modules/TensorFlowCompile.cmake
index 6f77174982f22..6c79e2f497845 100644
--- a/llvm/cmake/modules/TensorFlowCompile.cmake
+++ b/llvm/cmake/modules/TensorFlowCompile.cmake
@@ -29,11 +29,11 @@ endfunction()
 # Generate a mock model for tests.
 function(generate_mock_model model generate_mock_model_py config)
   tf_get_absolute_path(${model} ${CMAKE_CURRENT_BINARY_DIR} LLVM_ML_MODELS_ABSOLUTE)
-  tf_get_absolute_path(${generate_mock_model_py} ${CMAKE_CURRENT_SOURCE_DIR} GENERATE_INLINING_MODEL_ABSOLUTE)
+  tf_get_absolute_path(${generate_mock_model_py} ${CMAKE_CURRENT_SOURCE_DIR} GENERATED_MODEL_ABSOLUTE_PATH)
   tf_get_absolute_path(${config} ${CMAKE_CURRENT_SOURCE_DIR} LLVM_ML_MODEL_CONFIG_ABSOLUTE)
   message(WARNING "Autogenerated mock models should not be used in production builds.")
   execute_process(COMMAND python3
-    ${GENERATE_INLINING_MODEL_ABSOLUTE}
+    ${GENERATED_MODEL_ABSOLUTE_PATH}
     ${LLVM_ML_MODEL_CONFIG_ABSOLUTE}
     ${LLVM_ML_MODELS_ABSOLUTE}-autogenerated
     WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}