|
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 |
|
Definition at line 17 of file cxx11_tensor_custom_op.cpp.
DSizes<DenseIndex, 2> InsertZeros::dimensions |
( |
const Tensor< float, 2 > & |
input | ) |
const |
|
inline |
template<typename Output , typename Device >
void InsertZeros::eval |
( |
const Tensor< float, 2 > & |
input, |
|
|
Output & |
output, |
|
|
const Device & |
device |
|
) |
| const |
|
inline |
The documentation for this struct was generated from the following file: