Home
last modified time | relevance | path

Searched refs:_get_tensor_constant_from_node (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/
H A Dutils.py142 def _get_tensor_constant_from_node(node, m): function
234 conv_w = _get_tensor_constant_from_node(conv_weight_node, m)
235 conv_b = _get_tensor_constant_from_node(conv_bias_node, m)
242 bn_w = _get_tensor_constant_from_node(bn_args[1], m)
243 bn_b = _get_tensor_constant_from_node(bn_args[2], m)
244 bn_rm = _get_tensor_constant_from_node(bn_args[3], m)
245 bn_rv = _get_tensor_constant_from_node(bn_args[4], m)