Defined in File matx_util.cu
-
struct MatxUtil__MatxThreshold
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, double threshold, const ShapeType &stride)
TODO(Documentation)
-
template<typename InputT, std::enable_if_t<cudf::is_floating_point<InputT>()>* = nullptr>
inline void operator()(void *input_data, void *output_data, double threshold, const ShapeType &stride)
TODO(Documentation)
Public Members
- TensorIndex rows
- TensorIndex cols
- bool by_row
- rmm::cuda_stream_view stream
-
template<typename InputT, std::enable_if_t<!cudf::is_floating_point<InputT>()>* = nullptr>