source/tensor/core/arithmetic/Multiply.cpp · 72c9551b2cf517c6e8e0e37f8449f688f5a3a5b8 · NiuTrans / NiuTrans.Tensor · GitLab

Blame History Permalink

add CheckDev to check if two tensors are on the same device · 86cb0715
xiaotong committed Jun 12, 2019

86cb0715

Multiply.cpp 6.52 KB