Function caffe2_math::scale_with_alpha_from_pointer

source ·

pub fn scale_with_alpha_from_pointer<TAlpha, TData, Context>(
    n: i64,
    alpha: *const TAlpha,
    x: *const TData,
    y: *mut TData,
    context: *mut Context
)

Expand description

| Different from the Scale function above, if | alpha is passed in as a pointer, we will | assume that it lives on the Context device, | for example on GPU.