Public Member Functions | List of all members
BatchMatMul< TensorType > Struct Template Reference

Public Member Functions

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
 

Detailed Description

template<typename TensorType>
struct BatchMatMul< TensorType >

Definition at line 61 of file cxx11_tensor_custom_op.cpp.

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

Definition at line 62 of file cxx11_tensor_custom_op.cpp.

◆ dimensions() [2/2]

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

Definition at line 89 of file cxx11_tensor_custom_op_sycl.cpp.

◆ 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

Definition at line 71 of file cxx11_tensor_custom_op.cpp.

◆ 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

Definition at line 98 of file cxx11_tensor_custom_op_sycl.cpp.


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


gtsam
Author(s):
autogenerated on Fri Nov 1 2024 03:43:34