Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp610 std::unordered_set<Node*> scalar_nodes_to_rewrite; in ReplicateClampScalarArgs() local
619 scalar_nodes_to_rewrite.insert(n); in ReplicateClampScalarArgs()
628 for (Node* n : scalar_nodes_to_rewrite) { in ReplicateClampScalarArgs()
639 for (Node* n : scalar_nodes_to_rewrite) { in ReplicateClampScalarArgs()
643 for (Node* n : scalar_nodes_to_rewrite) { in ReplicateClampScalarArgs()