#include <TensorReduction.h>
Static Public Member Functions | |
template<typename InputDims , typename Index , size_t Rank> | |
static EIGEN_DEVICE_FUNC void | run (const InputDims &input_dims, const array< bool, Rank > &, Sizes<> *, array< Index, Rank > *reduced_dims) |
Definition at line 76 of file TensorReduction.h.
|
inlinestatic |
Definition at line 78 of file TensorReduction.h.