#include <TensorReduction.h>
Static Public Member Functions | |
template<typename InputDims , typename ReducedDims > | |
static EIGEN_DEVICE_FUNC void | run (const InputDims &input_dims, const array< bool, internal::array_size< InputDims >::value > &reduced, OutputDims *output_dims, ReducedDims *reduced_dims) |
Definition at line 56 of file TensorReduction.h.
|
inlinestatic |
Definition at line 58 of file TensorReduction.h.