mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
91797c0672
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10946 ``` codemod -d . --extensions cc,cpp,cu,cuh,h caffe2/proto/caffe2.pb.h caffe2/proto/caffe2_pb.h ``` Reviewed By: houseroad Differential Revision: D9539945 fbshipit-source-id: 497d04720e8e7e61c05ffe1b23733d0cb774de7e
22 lines
450 B
C++
22 lines
450 B
C++
#ifndef CAFFE2_TENSOR_INT8_H_
|
|
#define CAFFE2_TENSOR_INT8_H_
|
|
|
|
#include "caffe2/core/context.h"
|
|
#include "caffe2/core/tensor.h"
|
|
#include "caffe2/proto/caffe2_pb.h"
|
|
|
|
namespace caffe2 {
|
|
namespace int8 {
|
|
|
|
struct Int8TensorCPU {
|
|
float scale{1.0};
|
|
int32_t zero_point{0};
|
|
// Generally stores uint8_t data, but sometimes int32_t (e.g. bias
|
|
// parameters).
|
|
Tensor t{CPU};
|
|
};
|
|
} // namespace int8
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_TENSOR_INT8_H_
|