BatchMatMul Struct Reference

Public Member Functions
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

Detailed Description

Definition at line 61 of file cxx11_tensor_custom_op.cpp.

Member Function Documentation

DSizes<DenseIndex, 3> BatchMatMul::dimensions ( const Tensor< float, 3 > &  input1, const Tensor< float, 3 > &  input2  ) const

inline

Definition at line 62 of file cxx11_tensor_custom_op.cpp.

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

Definition at line 71 of file cxx11_tensor_custom_op.cpp.

---

The documentation for this struct was generated from the following file:

• cxx11_tensor_custom_op.cpp