Struct MatxUtil__MatxReduceMax

struct MatxUtil__MatxReduceMax

Public Functions

template<typename InputT, std::enable_if_t<!cudf::is_floating_point<InputT>()>* = nullptr>
inline void operator()(void *input_data, void *output_data)

template<typename InputT, std::enable_if_t<cudf::is_floating_point<InputT>()>* = nullptr>
inline void operator()(void *input_data, void *output_data)

template<typename InputT>
inline void reduce_rows(matx::tensor_t<InputT, 2> &input_tensor, matx::tensor_t<InputT, 2> &output_tensor, matx::index_t start, matx::index_t stop, matx::index_t output_idx)

Public Members

matx::index_t num_input_rows

matx::index_t num_output_rows

matx::index_t num_cols

std::vector<matx::index_t> input_stride

const ShapeType &seq_ids

TensorIndex seq_id_offset

rmm::cuda_stream_view stream

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