set_handle_precision

Function set_handle_precision 

Source
pub fn set_handle_precision(
    handle: &GpuTensorHandle,
    precision: ProviderPrecision,
)
Expand description

Record the precision associated with a GPU tensor handle so host operations can reconstruct the original dtype when gathering back to the CPU.