C++ (Cpp) s_native_addmmの例

コード例 #1

ファイル: LegacyBridge.cpp プロジェクト: SiddharthTiwari/pytorch

Tensor addmm(const Tensor& self, const Tensor& mat1, const Tensor& mat2, Scalar beta, Scalar alpha) {
  // See Note [Multiple dispatch to sparse]
  auto mat1_sparse = mat1.is_sparse();
  if (mat1_sparse) {
    Tensor b_self;
    std::tie(b_self) = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm");
    return s_native_addmm(b_self, mat1, mat2, beta, alpha);
  } else {
    return legacy::th::_th_addmm(self, mat1, mat2, beta, alpha);
  }
}

コード例 #2

ファイル: LegacyBridge.cpp プロジェクト: gtgalone/pytorch

Tensor addmm(const Tensor& self, const Tensor& mat1, const Tensor& mat2, Scalar beta, Scalar alpha) {
  if (!self.is_cuda()) {
    // See Note [CPU sparse is globally native] and Note [Multiple dispatch to sparse]
    auto mat1_sparse = mat1.is_sparse();
    if (mat1_sparse) {
      Tensor b_self;
      std::tie(b_self) = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm");
      return s_native_addmm(b_self, mat1, mat2, beta, alpha);
    } else {
      return th_addmm(self, mat1, mat2, beta, alpha);
    }
  } else {
    return th_addmm(self, mat1, mat2, beta, alpha);
  }
}