Function caffe2_math::broadcast_impl
source · pub fn broadcast_impl<T>(
x_ndim: i32,
x_dims: *const i32,
y_ndim: i32,
y_dims: *const i32,
alpha: T,
x: *const T,
y: *mut T,
context: *mut CPUContext
)
Expand description
| Common math functions being used in Caffe that | do not have a BLAS or MKL equivalent. For all | these functions, we will simply implement them | either via Eigen or via custom code.