MagmaDNN  1.0
c++NeuralNetworkFramework
productop.h File Reference
#include <vector>
#include "compute/operation.h"
#include "compute/variable.h"
#include "tensor/tensor.h"
#include "utilities_internal.h"
#include "compute/product/product_internal.h"
#include "math/scalar_tensor_product.h"
Include dependency graph for productop.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  magmadnn::op::ProductOp< T >
 

Enumerations

enum  product_op_t { SCALAR_PROD_TENSOR, TENSOR_PROD_SCALAR, TENSOR_PROD_TENSOR }
 

Functions

template<typename T >
ProductOp< T > * magmadnn::op::product (Operation< T > *a, Operation< T > *b, bool copy, bool needs_grad)
 
template<typename T >
ProductOp< T > * magmadnn::op::product (T alpha, Operation< T > *a, Operation< T > *b, bool copy, bool needs_grad)
 

Detailed Description

Author
Daniel Nichols
Version
0.1
Date
2019-05-21