source/tensor/core/arithmetic/DivDim.cpp · 979698b4c1c32ea00b3a812215b2bdc4a6dc945a · 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

DivDim.cpp 5.27 KB