MagmaDNN  1.0
c++NeuralNetworkFramework
product_internal.h
Go to the documentation of this file.
1 
9 #pragma once
10 
11 #include "tensor/tensor.h"
12 
13 namespace magmadnn {
14 namespace internal {
15 
16 template <typename T>
17 void product_full(T alpha, Tensor<T> *a, Tensor<T> *b, Tensor<T> *out);
18 
19 
20 #if defined(_HAS_CUDA_)
21 template <typename T>
22 void product_full_device(T alpha, Tensor<T> *a, Tensor<T> *b, Tensor<T> *out);
23 #endif
24 
25 
26 template <typename T>
27 void scalar_tensor_product_full(T alpha, Tensor<T> *a, Tensor<T> *out);
28 
29 
30 #if defined(_HAS_CUDA_)
31 template <typename T>
32 void scalar_tensor_product_full_device(T alpha, Tensor<T> *a, Tensor<T> *out);
33 #endif
34 
35 
36 } // namespace internal
37 } // namespace magmadnn
magmadnn
Definition: addop.cpp:11