Searched refs:scale_node (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/_export/passes/ |
| D | replace_quantized_ops_with_standard_ops_pass.py | 46 scale_node: Union[float, torch.fx.Node],
57 scale_node,
102 scale_node: Union[float, torch.fx.Node],
115 scale_node,
127 scale_node,
298 scale_node, zero_point_node = node.args[2], node.args[3]
346 return op_res_node, scale_node, zero_point_node
351 scale_node, zero_point_node = node.args[2], node.args[3]
383 return op_res_node, scale_node, zero_point_node
406 scale_node, zero_point_node = args[-2], args[-1]
[all …]
|
| /external/executorch/backends/xnnpack/operators/ |
| D | op_sdpa.py | 75 scale_node = self.get_fake_attr("scale", t)
77 scale_node,
95 scale_id = vals_to_ids[scale_node]
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | _lower_to_native_backend.py | 674 (_, scale_node, zero_point_node, _) = q_node.args
677 assert isinstance(scale_node, Node)
690 output_scale = getattr(model, scale_node.target) # type: ignore[arg-type]
704 model.graph.erase_node(scale_node)
732 (_, scale_node, zero_point_node, _) = q_node.args
735 assert isinstance(scale_node, Node)
750 output_scale = getattr(model, scale_node.target) # type: ignore[arg-type]
768 model.graph.erase_node(scale_node)
1098 (_, scale_node, zero_point_node, _) = q_node.args
1122 qop_node_args.extend([scale_node, zero_point_node])
[all …]
|
| D | convert.py | 284 scale_node = graph.create_node(
294 "_scale_": scale_node,
|
| /external/executorch/backends/xnnpack/utils/ |
| D | quant_utils.py | 153 scale_node = node.args[2]
155 args = [input_node, scale_node, zero_point_node]
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | utils.py | 178 scale_node = get_normalized_nth_input(node, gm, scale_arg_idx)
180 assert isinstance(scale_node, Node) and isinstance(scale_node.target, str)
182 scale_obj = getattr_from_fqn(gm, scale_node.target)
|
| D | n_shadows_utils.py | 1021 _weight, scale_node, zp_node, axis, dtype = quant_node.args
1022 scale_val = getattr_from_fqn(shadow_wrapper, scale_node.target)
1027 _weight, scale_node, zp_node, dtype = quant_node.args
1028 scale_val = getattr_from_fqn(shadow_wrapper, scale_node.target)
|
| D | graph_passes.py | 214 scale_node = graph_c.create_node(
229 (prev_node_c, scale_node, zero_point_node, torch.quint8),
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | port_metadata_pass.py | 87 scale_node = choose_qparam_users.pop()
88 dynamic_q_node = next(iter(scale_node.users.keys()))
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | unpack_quantized_weights.cpp | 166 Node* scale_node = graph->create(prim::Constant); in CreateQuantizedWeights() local
171 scale_node->t_(Symbol::attr("value"), scale_value.clone()); in CreateQuantizedWeights()
191 return {data_node, scale_node, zero_point_node, axis_node}; in CreateQuantizedWeights()
|
| /external/pytorch/test/onnx/ |
| D | test_utility_funs.py | 1935 scale_node = [
1938 self.assertEqual(len(scale_node), 1)
1939 self.assertEqual(scale_node[0].op_type, "Constant")
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 2647 auto* scale_node = in AddMklLayerNorm() local
2657 fused_node.add_input(scale_node->name()); in AddMklLayerNorm()
|