Defined in File matx_util.cu
-
struct MatxUtil__MatxLogits
TODO(Documentation)
Public Functions
-
template<typename InputT, std::enable_if_t<!cudf::is_floating_point<InputT>()>* = nullptr>
inline void operator()(void *input_data, void *output_data) TODO(Documentation)
-
template<typename InputT, std::enable_if_t<cudf::is_floating_point<InputT>()>* = nullptr>
inline void operator()(void *input_data, void *output_data) TODO(Documentation)
Public Members
-
size_t element_count
-
rmm::cuda_stream_view stream
-
template<typename InputT, std::enable_if_t<!cudf::is_floating_point<InputT>()>* = nullptr>