Caffe2 - C++ API
A deep learning, cross platform ML framework
typed_axpy.h
1 
17 #pragma once
18 
19 namespace caffe2 {
20 
21 // Similar to Axpy that calculate y = a * x + y, but allowing x and y to be
22 // of different data types.
23 // It also provides a performance optimization hint (use_a) to see if a is going
24 // to be 1 or not.
25 template <typename IN, typename OUT>
26 void TypedAxpy(int N, const OUT a, const IN* x, OUT* y);
27 
28 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.