pub struct DequantizePerGroupArgs<'a, TIn: Element, TOut: IntElement> {
pub input: TensorRef<'a, TOut, 2>,
pub scale: TensorRef<'a, TIn, 2>,
pub zero_point: TensorRef<'a, i32, 2>,
pub output: TensorMut<'a, TIn, 2>,
}Expand description
Args for the dequant-per-group launch.
Fields§
§input: TensorRef<'a, TOut, 2>Quantized input [outer, axis_size] in int.
scale: TensorRef<'a, TIn, 2>Per-group scale [outer, num_groups] in FP.
zero_point: TensorRef<'a, i32, 2>Per-group zero-point [outer, num_groups] in i32.
output: TensorMut<'a, TIn, 2>Output [outer, axis_size] in FP.
Auto Trait Implementations§
impl<'a, TIn, TOut> !UnwindSafe for DequantizePerGroupArgs<'a, TIn, TOut>
impl<'a, TIn, TOut> Freeze for DequantizePerGroupArgs<'a, TIn, TOut>
impl<'a, TIn, TOut> RefUnwindSafe for DequantizePerGroupArgs<'a, TIn, TOut>where
TOut: RefUnwindSafe,
TIn: RefUnwindSafe,
impl<'a, TIn, TOut> Send for DequantizePerGroupArgs<'a, TIn, TOut>
impl<'a, TIn, TOut> Sync for DequantizePerGroupArgs<'a, TIn, TOut>
impl<'a, TIn, TOut> Unpin for DequantizePerGroupArgs<'a, TIn, TOut>
impl<'a, TIn, TOut> UnsafeUnpin for DequantizePerGroupArgs<'a, TIn, TOut>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more