TrainingExtensions/tensorflow/src/python/aimet_tensorflow/quantsim_straight_through_grad.py
def _qc_straight_through_estimator_grad(op, grad):
    # pylint: disable=unused-argument
    """
    straight through estimator logic used to compute gradient for Quantize Op.
    :param op: quantize op
    :param grad: gradient
    :return: gradients computed per input
    """

    dloss_by_dx = _compute_dloss_by_dx(op, grad)
    return dloss_by_dx, None, None, None, None, None, None

    straight through estimator logic used to compute gradient for Quantize Op.
    :param op: quantize op
    :param grad: gradient
    :return: gradients computed per input
    
straight through estimator logic used to compute gradient for Quantize Op.
straight through estimator logic used to compute gradient for Quantize Op .
def _qc_straight_through_estimator_grad(op, grad):
    dloss_by_dx = _compute_dloss_by_dx(op, grad)
    return dloss_by_dx, None, None, None, None, None, None
def _qc_straight_through_estimator_grad ( op , grad ) : dloss_by_dx = _compute_dloss_by_dx ( op , grad ) return dloss_by_dx , None , None , None , None , None , None
straight through estimator logic used to compute gradient for Quantize Op.
straight through estimator logic used to compute gradient for Quantize Op .
# pylint: disable=unused-argument """
    straight through estimator logic used to compute gradient for Quantize Op.
    :param op: quantize op
    :param grad: gradient
    :return: gradients computed per input
    """
gradients computed per input
gradients computed per input
op
None
grad
None