Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dnormalize_ops.cpp15 iter->replaceWithNewSymbol(alias->second); in normalizeOpAliases()
27 Node* newSub = iter->replaceWithNewSymbol(aten::sub); in normalizeRSub()
43 iter->replaceWithNewSymbol(aten::eq); in normalizeIsBool()
48 iter->replaceWithNewSymbol(aten::ne); in normalizeIsBool()
Drestore_mutation.cpp67 Node* inplace_node = node->replaceWithNewSymbol( in FunctionalToInplace()
Dfrozen_ops_to_mkldnn.cpp220 node->replaceWithNewSymbol( in InplaceMKLDNNSubgraph()
768 body_node->replaceWithNewSymbol(Symbol::prim("MKLDNNScalarMul")); in ComputeSubgraphInMKLDNN()
775 body_node->replaceWithNewSymbol(Symbol::prim("MKLDNNLayerNorm")); in ComputeSubgraphInMKLDNN()
781 body_node->replaceWithNewSymbol(prim::MKLDNNHardSwish); in ComputeSubgraphInMKLDNN()
787 body_node->replaceWithNewSymbol(prim::MKLDNNHardSigmoid); in ComputeSubgraphInMKLDNN()
817 body_node->replaceWithNewSymbol(Symbol::prim("mkldnn_convolution")); in ComputeSubgraphInMKLDNN()
/external/pytorch/torch/csrc/jit/ir/
Dir.h495 Node* replaceWithNewSymbol(Symbol new_symbol);
Dir.cpp1362 Node* Node::replaceWithNewSymbol(Symbol new_symbol) { in replaceWithNewSymbol() function in torch::jit::Node