|
DSizes< DenseIndex, 3 > | dimensions (const Tensor< float, 3 > &input1, const Tensor< float, 3 > &input2) const |
|
DSizes< DenseIndex, 3 > | dimensions (const TensorType &input1, const TensorType &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 |
|
template<typename Output , typename Device > |
void | eval (const TensorType &input1, const TensorType &input2, Output &output, const Device &device) const |
|
template<typename TensorType>
struct BatchMatMul< TensorType >
Definition at line 61 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 BatchMatMul< TensorType >::eval |
( |
const Tensor< float, 3 > & |
input1, |
|
|
const Tensor< float, 3 > & |
input2, |
|
|
Output & |
output, |
|
|
const Device & |
device |
|
) |
| const |
|
inline |
◆ eval() [2/2]
template<typename TensorType >
template<typename Output , typename Device >
void BatchMatMul< TensorType >::eval |
( |
const TensorType & |
input1, |
|
|
const TensorType & |
input2, |
|
|
Output & |
output, |
|
|
const Device & |
device |
|
) |
| const |
|
inline |
The documentation for this struct was generated from the following files: