diff --git a/llvm/projects/keras/frontend/hpvm_dfg_translator.py b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
index 233652600a564f2ca2533030b1f623b4aab06f48..eda8ed9da8ab9e34f77a6165ca79d2e0a5e4e4f8 100644
--- a/llvm/projects/keras/frontend/hpvm_dfg_translator.py
+++ b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
@@ -166,7 +166,7 @@ class HPVMTranslator:
         node_header_str += ", "
         
     node_header_str += ") { \n" 
-    node_header_str += "  __visc__hint(visc::CUDNN_TARGET); \n"
+    node_header_str += "  __visc__hint(visc::PROMISE_TARGET); \n"
     node_header_str += "  __visc__attributes(" + str(num_params) + ", "
 
     for i in range(num_params):
@@ -462,7 +462,7 @@ class HPVMTranslator:
     if layer_type == "MaxPooling2D":
       func_name = "__visc__tensor_pool_max"     
     if layer_type == "AveragePooling2D":
-      func_name = "__visc__tensor_pool_avg"
+      func_name = "__visc__tensor_pool_mean"
       
     inst_str += "  void* r = " + func_name + "(t1, "
     inst_str += str(pool_size[0]) + ", " + str(pool_size[1]) + ", "