TR-mbed 1.0
Loading...
Searching...
No Matches
Public Member Functions | List of all members
InsertZeros< TensorType > Struct Template Reference

Public Member Functions

DSizes< DenseIndex, 2 > dimensions (const Tensor< float, 2 > &input) const
 
template<typename Output , typename Device >
void eval (const Tensor< float, 2 > &input, Output &output, const Device &device) const
 
DSizes< DenseIndex, 2 > dimensions (const TensorType &input) const
 
template<typename Output , typename Device >
void eval (const TensorType &input, Output &output, const Device &device) const
 

Member Function Documentation

◆ dimensions() [1/2]

template<typename TensorType >
DSizes< DenseIndex, 2 > InsertZeros< TensorType >::dimensions ( const Tensor< float, 2 > &  input) const
inline

◆ dimensions() [2/2]

template<typename TensorType >
DSizes< DenseIndex, 2 > InsertZeros< TensorType >::dimensions ( const TensorType &  input) const
inline

◆ eval() [1/2]

template<typename TensorType >
template<typename Output , typename Device >
void InsertZeros< TensorType >::eval ( const Tensor< float, 2 > &  input,
Output &  output,
const Device &  device 
) const
inline

◆ eval() [2/2]

template<typename TensorType >
template<typename Output , typename Device >
void InsertZeros< TensorType >::eval ( const TensorType &  input,
Output &  output,
const Device &  device 
) const
inline

The documentation for this struct was generated from the following files: