Skip to content
Snippets Groups Projects
Commit a8af6c46 authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Fixing pool_mean translation bug

parent 80211d08
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
llvm/projects/keras/frontend/hpvm_dfg_translator.py
+ 2
− 2
View file @ a8af6c46
...@@ -166,7 +166,7 @@ class HPVMTranslator: ...@@ -166,7 +166,7 @@ class HPVMTranslator:
node_header_str += ", " node_header_str += ", "
node_header_str += ") { \n" 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) + ", " node_header_str += " __visc__attributes(" + str(num_params) + ", "
for i in range(num_params): for i in range(num_params):
...@@ -462,7 +462,7 @@ class HPVMTranslator: ...@@ -462,7 +462,7 @@ class HPVMTranslator:
if layer_type == "MaxPooling2D": if layer_type == "MaxPooling2D":
func_name = "__visc__tensor_pool_max" func_name = "__visc__tensor_pool_max"
if layer_type == "AveragePooling2D": 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 += " void* r = " + func_name + "(t1, "
inst_str += str(pool_size[0]) + ", " + str(pool_size[1]) + ", " inst_str += str(pool_size[0]) + ", " + str(pool_size[1]) + ", "
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment