|
DSizes< DenseIndex, 3 > | dimensions (const Tensor< float, 3 > &input1, const Tensor< float, 3 > &input2) const |
|
template<typename Output , typename Device > |
void | eval (const Tensor< float, 3 > &input1, const Tensor< float, 3 > &input2, Output &output, const Device &device) const |
|
Definition at line 61 of file cxx11_tensor_custom_op.cpp.
DSizes<DenseIndex, 3> BatchMatMul::dimensions |
( |
const Tensor< float, 3 > & |
input1, |
|
|
const Tensor< float, 3 > & |
input2 |
|
) |
| const |
|
inline |
template<typename Output , typename Device >
void BatchMatMul::eval |
( |
const Tensor< float, 3 > & |
input1, |
|
|
const Tensor< float, 3 > & |
input2, |
|
|
Output & |
output, |
|
|
const Device & |
device |
|
) |
| const |
|
inline |
The documentation for this struct was generated from the following file: