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 |
Definition at line 17 of file cxx11_tensor_custom_op.cpp.
|
inline |
Definition at line 18 of file cxx11_tensor_custom_op.cpp.
|
inline |
Definition at line 26 of file cxx11_tensor_custom_op.cpp.