|
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 |
|
template<typename TensorType>
struct InsertZeros< TensorType >
Definition at line 17 of file cxx11_tensor_custom_op.cpp.
◆ dimensions() [1/2]
template<typename TensorType >
◆ dimensions() [2/2]
template<typename TensorType >
◆ 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: