InsertZeros< TensorType > Struct Template Reference

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

Detailed Description

template<typename TensorType>
struct InsertZeros< TensorType >

Definition at line 17 of file cxx11_tensor_custom_op.cpp.

Member Function Documentation

dimensions() [1/2]

template<typename TensorType >

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

inline

Definition at line 18 of file cxx11_tensor_custom_op.cpp.

dimensions() [2/2]

template<typename TensorType >

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

inline

Definition at line 26 of file cxx11_tensor_custom_op_sycl.cpp.

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

Definition at line 26 of file cxx11_tensor_custom_op.cpp.

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

Definition at line 34 of file cxx11_tensor_custom_op_sycl.cpp.

---

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

• cxx11_tensor_custom_op.cpp
• cxx11_tensor_custom_op_sycl.cpp