CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
< Layout function
#include <tensor_compare.h>
Public Member Functions | |
TensorContainsFunc () | |
Ctor. More... | |
TensorContainsFunc (TensorView< Element, Layout > const &view_, Element value_) | |
Ctor. More... | |
void | operator() (Coord< Layout::kRank > const &coord) |
Visits a coordinate. More... | |
operator bool () const | |
Returns true if equal. More... | |
Public Attributes | |
TensorView< Element, Layout > | view |
Element | value |
bool | contains |
Coord< Layout::kRank > | location |
|
inline |
|
inline |
|
inline |
|
inline |
bool cutlass::reference::host::detail::TensorContainsFunc< Element, Layout >::contains |
Coord<Layout::kRank> cutlass::reference::host::detail::TensorContainsFunc< Element, Layout >::location |
Element cutlass::reference::host::detail::TensorContainsFunc< Element, Layout >::value |
TensorView<Element, Layout> cutlass::reference::host::detail::TensorContainsFunc< Element, Layout >::view |