Defined in File matx_util.cu
-
struct MatxUtil__MatxCreateSegIds
TODO(Documentation)
Public Functions
-
template<typename OutputT, std::enable_if_t<!std::is_integral_v<OutputT>>* = nullptr>
inline void operator()(void *output_data)
TODO(Documentation)
-
template<typename OutputT, std::enable_if_t<std::is_integral_v<OutputT>>* = nullptr>
inline void operator()(void *output_data)
TODO(Documentation)
Public Members
- TensorIndex start_idx
- TensorIndex element_count
- TensorIndex fea_len
- rmm::cuda_stream_view stream
-
template<typename OutputT, std::enable_if_t<!std::is_integral_v<OutputT>>* = nullptr>