Function set_variable_data

python/paddle/static/quantization/utils.py:126–136

Set the value of var node by name, if the node exits,

(scope, place, var_name, np_value)

def set_variable_data(scope, place, var_name, np_value):
    '''
    Set the value of var node by name, if the node exits,
    '''
    assert isinstance(np_value, np.ndarray), (
        'The type of value should be numpy array.'
    )
    var_node = scope.find_var(var_name)
    if var_node is not None:
        tensor = var_node.get_tensor()
        tensor.set(np_value, place)


def quant_tensor(x, scale, quant_axis=0, weight_bits=8, onnx_format=False):