/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/sgd
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/sgd/math_lp.h (655B)
#pragma once
#ifdef __AVX__
#include
#endif
#include "caffe2/utils/math.h"
namespace caffe2 {
namespace internal {
// Z=X*Y
template
void dot(const int N, const XT* x, const YT* y, ZT* z, CPUContext* ctx) {
CAFFE_THROW("Unsupported, see specialized implementations");
}
template <>
void dot(
const int N,
const float* x,
const float* y,
float* z,
CPUContext* ctx);
template <>
void dot(
const int N,
const float* x,
const at::Half* y,
float* z,
CPUContext* ctx);
} // namespace internal
} // namespace caffe2