Struct MatxUtil__MatxLogits

(Latest Version)
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

© Copyright 2023, NVIDIA. Last updated on Apr 11, 2023.