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
- TensorIndex element_count
- rmm::cuda_stream_view stream
-
template<typename InputT, std::enable_if_t<!cudf::is_floating_point<InputT>()>* = nullptr>