/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_thresholding.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_thresholding.hpp Source File#

2 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_thresholding.hpp Source File

include

Go to the documentation of this file.

  
 #pragma once
  
 #include <operator_types.h>
  
 #include <i_operator.hpp>
  
 #include "core/tensor.hpp"
  
  
 namespace roccv {
 class Threshold final : public IOperator {
    public:
     Threshold(eThresholdType threshType, int32_t maxBatchSize);
  
     ~Threshold();
     void operator()(hipStream_t stream, const roccv::Tensor& input, const roccv::Tensor& output,
                     const roccv::Tensor& thresh, const roccv::Tensor& maxVal, eDeviceType device = eDeviceType::GPU);
  
    private:
     eThresholdType m_threshType;
     int32_t m_maxBatchSize;
 };
 }  // namespace roccv