Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::OpWrapper< OpType, T > Class Template Reference

Wrap a floating-point operator with quantized inputs with type T. More...

#include <op_wrapper.h>

Public Member Functions

 OpWrapper (OperatorBase *op, dnnlowp::QuantizationFactory *qfactory)
 
void DequantizeInput ()
 
OpType * Get ()
 
dnnlowp::TensorQuantizationParams GetOutputQuantizationParams (dnnlowp::QuantizationFactory *qfactory, int index=0)
 

Detailed Description

template<typename OpType, typename T>
class caffe2::OpWrapper< OpType, T >

Wrap a floating-point operator with quantized inputs with type T.

This class is to measure quantization error against fp32 reference.

Definition at line 15 of file op_wrapper.h.


The documentation for this class was generated from the following file: