Skip to content

GitLab

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
Show 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