FM / utils

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils

Name	Size	Mode	Actions
math/	-	0755	rm
threadpool/	-	0755	rm
bench_utils.h	854	0644	edit dl rm
cast.h	1110	0644	edit dl rm
cblas.h	34183	0644	edit dl rm
conversions.h	661	0644	edit dl rm
cpuid.h	2560	0644	edit dl rm
cpu_neon.h	1578	0644	edit dl rm
eigen_utils.h	6706	0644	edit dl rm
filler.h	3987	0644	edit dl rm
fixed_divisor.h	3634	0644	edit dl rm
knobs.h	532	0644	edit dl rm
knob_patcher.h	715	0644	edit dl rm
map_utils.h	635	0644	edit dl rm
math-detail.h	1916	0644	edit dl rm
math.h	15251	0644	edit dl rm
murmur_hash3.h	909	0644	edit dl rm
proto_convert.h	263	0644	edit dl rm
proto_utils.h	13254	0644	edit dl rm
proto_wrap.h	378	0644	edit dl rm
signal_handler.h	698	0644	edit dl rm
simple_queue.h	2602	0644	edit dl rm
smart_tensor_printer.h	1402	0644	edit dl rm
string_utils.h	1206	0644	edit dl rm
zmq_helper.h	3020	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/math-detail.h (1916B)

#ifndef CAFFE2_UTILS_MATH_DETAIL_H_
#define CAFFE2_UTILS_MATH_DETAIL_H_
namespace caffe2 {

class CPUContext;

namespace math {
namespace detail {

// proxy to a class because of partial specialization limitations for functions

template<typename T, class Context, int FixedSize>
struct ScaleImpl {
  inline void operator()(
      const int N,
      const float alpha,
      const T* x,
      T* y,
      Context* context) {
    Scale(N, alpha, x, y, context);
  }
};

// Put light-weight implementations in .h file to enable inlining
template<typename T>
struct ScaleImpl<T, CPUContext, 1> {
  inline void operator()(
      const int N,
      const float alpha,
      const T* x,
      T* y,
      CPUContext* /*context*/) {
    DCHECK_EQ(N, 1);
    *y = *x * alpha;
  }
};

template<typename T, class Context, int FixedSize>
struct AxpyImpl {
  inline void operator()(
      const int N,
      const float alpha,
      const T* x,
      T* y,
      Context* context) {
    Axpy(N, alpha, x, y, context);
  }
};

// Put light-weight implementations in .h file to enable inlining
template<typename T>
struct AxpyImpl<T, CPUContext, 1> {
  inline void operator()(
      const int N,
      const float alpha,
      const T* x,
      T* y,
      CPUContext* /*context*/) {
    DCHECK_EQ(N, 1);
    *y += *x * alpha;
  }
};

}  // namespace detail

template <typename T, class Context, int FixedSize>
inline void ScaleFixedSize(
    const int N,
    const float alpha,
    const T* x,
    T* y,
    Context* context) {
  detail::ScaleImpl<T, Context, FixedSize>()(N, alpha, x, y, context);
}

template <typename T, class Context, int FixedSize>
inline void AxpyFixedSize(
    const int N,
    const float alpha,
    const T* x,
    T* y,
    Context* context) {
  detail::AxpyImpl<T, Context, FixedSize>()(N, alpha, x, y, context);
}

}  // namespace math
}  // namespace caffe2

#endif  // CAFFE2_UTILS_MATH_DETAIL_H_

#ifndef CAFFE2_UTILS_MATH_DETAIL_H_ #define CAFFE2_UTILS_MATH_DETAIL_H_ namespace caffe2 { class CPUContext; namespace math { namespace detail { // proxy to a class because of partial specialization limitations for functions template struct ScaleImpl { inline void operator()( const int N, const float alpha, const T* x, T* y, Context* context) { Scale(N, alpha, x, y, context); } }; // Put light-weight implementations in .h file to enable inlining template struct ScaleImpl { inline void operator()( const int N, const float alpha, const T* x, T* y, CPUContext* /*context*/) { DCHECK_EQ(N, 1); *y = *x * alpha; } }; template struct AxpyImpl { inline void operator()( const int N, const float alpha, const T* x, T* y, Context* context) { Axpy(N, alpha, x, y, context); } }; // Put light-weight implementations in .h file to enable inlining template struct AxpyImpl { inline void operator()( const int N, const float alpha, const T* x, T* y, CPUContext* /*context*/) { DCHECK_EQ(N, 1); *y += *x * alpha; } }; } // namespace detail template inline void ScaleFixedSize( const int N, const float alpha, const T* x, T* y, Context* context) { detail::ScaleImpl()(N, alpha, x, y, context); } template inline void AxpyFixedSize( const int N, const float alpha, const T* x, T* y, Context* context) { detail::AxpyImpl()(N, alpha, x, y, context); } } // namespace math } // namespace caffe2 #endif // CAFFE2_UTILS_MATH_DETAIL_H_