pytorch_quantization.tensor_quant.FakeTensorQuantFunction Class Reference

Inheritance diagram for pytorch_quantization.tensor_quant.FakeTensorQuantFunction:

Collaboration diagram for pytorch_quantization.tensor_quant.FakeTensorQuantFunction:

Static Public Member Functions
def	forward (ctx, inputs, amax, num_bits=8, unsigned=False, narrow_range=True)
def	backward (ctx, grad_outputs)

Detailed Description

Fake version of TensorQuantFunction
See comments of TensorQuantFunction, arguments are the same.

Member Function Documentation

forward()

def pytorch_quantization.tensor_quant.FakeTensorQuantFunction.forward (   ctx,   inputs,   amax,   num_bits = 8,   unsigned = False,   narrow_range = True  )

static

Here is the call graph for this function:

backward()

def pytorch_quantization.tensor_quant.FakeTensorQuantFunction.backward (   ctx,   grad_outputs  )

static

---

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

• tensor_quant.py