Public Member Functions | List of all members
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:


hebiros
Author(s): Xavier Artache , Matthew Tesch
autogenerated on Thu Sep 3 2020 04:09:50