Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "af13a90ebdcdc240d6cb159ffbe8f7661f20828d"
Unverified Commit d70beba7 authored by Lu Lu's avatar Lu Lu Committed by GitHub
Browse files

Bug fix: change th to tf (#3182) 


Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
parent c8d4d6fb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/backend/tensorflow/tensor.py
View file @ d70beba7
......@@ -188,7 +188,7 @@ def mean(input, dim):
def reduce_mean(input):
return th.reduce_mean(input)
return tf.reduce_mean(input)
def max(input, dim):
......
Markdown is supported
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