Struct Threshold 

pub struct Threshold { /* private fields */ }

Threshold operator (binary / trunc / tozero / Otsu / triangle).

Implementations

impl Threshold

pub fn new(threshold_type: u32, max_batch_size: i32) -> Result<Self>

Wraps cvcudaThresholdCreate. threshold_type selects between binary / truncate / tozero / Otsu / triangle; max_batch_size caps the per-call batch the operator will accept.

pub unsafe fn submit( &self, stream: *mut c_void, input: &Tensor, output: &Tensor, thresh: &Tensor, maxval: &Tensor, ) -> Result<()>

Safety

stream live.

Trait Implementations

impl Debug for Threshold

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.