MagmaDNN  1.0
c++NeuralNetworkFramework
scalar_tensor_product.h
Go to the documentation of this file.
1 
9 #pragma once
10 
11 #include "tensor/tensor.h"
12 
13 namespace magmadnn {
14 namespace math {
15 
16 template <typename T>
17 void scalar_tensor_product(T scalar, Tensor<T> *x, Tensor<T> *out);
18 
19 #if defined(_HAS_CUDA_)
20 template <typename T>
21 void scalar_tensor_product_device(T scalar, Tensor<T> *x, Tensor<T> *out);
22 #endif
23 
24 }
25 }
Definition: addop.cpp:11