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