Public Member Functions | List of all members
InsertZeros Struct 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
 

Detailed Description

Definition at line 17 of file cxx11_tensor_custom_op.cpp.

Member Function Documentation

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

Definition at line 18 of file cxx11_tensor_custom_op.cpp.

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

Definition at line 26 of file cxx11_tensor_custom_op.cpp.


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


hebiros
Author(s): Xavier Artache , Matthew Tesch
autogenerated on Thu Sep 3 2020 04:09:50