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
}
magmadnn
Definition:
addop.cpp:11
tensor.h
include
math
scalar_tensor_product.h
Generated by
1.8.13