cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc > Struct Template Reference

Helper to apply a binary operator in place.

#include <tensor_elementwise.h>

Collaboration diagram for cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >:

Public Member Functions
	TensorFuncBinaryOp ()
	Constructor. More...
	TensorFuncBinaryOp (TensorView< ElementD, LayoutD > const &view_d_, TensorRef< ElementA, LayoutA > const &ref_a_, TensorRef< ElementB, LayoutB > const &ref_b_, BinaryFunc func=BinaryFunc())
	Constructor. More...
void	operator() (Coord< LayoutD::kRank > const &coord) const
	Equality check. More...

Public Attributes
TensorView< ElementD, LayoutD >	view_d
	View of left-hand-side tensor. More...
TensorRef< ElementA, LayoutA >	ref_a
TensorRef< ElementB, LayoutB >	ref_b
BinaryFunc	func

Constructor & Destructor Documentation

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::TensorFuncBinaryOp ( )

inline

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::TensorFuncBinaryOp ( TensorView< ElementD, LayoutD > const &  view_d_, TensorRef< ElementA, LayoutA > const &  ref_a_, TensorRef< ElementB, LayoutB > const &  ref_b_, BinaryFunc  func = BinaryFunc()  )

inline

Member Function Documentation

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

void cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::operator() ( Coord< LayoutD::kRank > const &  coord ) const

inline

Member Data Documentation

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

BinaryFunc cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::func

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

TensorRef<ElementA, LayoutA> cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::ref_a

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

TensorRef<ElementB, LayoutB> cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::ref_b

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementD , typename LayoutD , typename BinaryFunc >

TensorView<ElementD, LayoutD> cutlass::reference::host::detail::TensorFuncBinaryOp< ElementA, LayoutA, ElementB, LayoutB, ElementD, LayoutD, BinaryFunc >::view_d

---

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

• host/tensor_elementwise.h