BatchMatMul< TensorType > Struct Template 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
DSizes< DenseIndex, 3 >	dimensions (const TensorType &input1, const TensorType &input2) const
template<typename Output , typename Device >
void	eval (const TensorType &input1, const TensorType &input2, Output &output, const Device &device) const

Member Function Documentation

dimensions() [1/2]

template<typename TensorType >

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

inline

dimensions() [2/2]

template<typename TensorType >

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

inline

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:

• core/util/algorithms/eigen-3.4.0/unsupported/test/cxx11_tensor_custom_op.cpp
• core/util/algorithms/eigen-3.4.0/unsupported/test/cxx11_tensor_custom_op_sycl.cpp