Home
last modified time | relevance | path

Searched defs:child_block (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/onnx/
Dpeephole.cpp127 for (auto* child_block : n->blocks()) { in fuseBroadcast() local
184 for (auto* child_block : n->blocks()) { in fuseConsecutiveTransposes() local
207 for (auto* child_block : n->blocks()) { in eliminateNopTranspose() local
224 for (auto* child_block : n->blocks()) { in fuseTransposeIntoGemm() local
263 for (auto* child_block : n->blocks()) { in pushPackingPastRnn() local
402 for (auto* child_block : n->blocks()) { in removeNopPacking() local
433 for (auto* child_block : n->blocks()) { in hackFixupPadPackedShapes() local
541 for (auto* child_block : n->blocks()) { in fixDefaultRnnHiddenState() local
560 for (auto* child_block : n->blocks()) { in fixDefaultLstmCellState() local
755 for (auto* child_block : it->blocks()) { in fuseListConstructListUnpack() local
[all …]
Dpreprocess_for_onnx.cpp89 for (auto* child_block : it->blocks()) { in FuseWithListUnpack() local
139 for (auto* child_block : it->blocks()) { in ReplaceAddWithConcat() local
190 for (auto* child_block : it->blocks()) { in fuseListAndListUnpack() local
Deval_peephole.cpp44 for (auto* child_block : it->blocks()) { in fuseConvBatchNorm() local
Dremove_inplace_ops_for_onnx.cpp331 for (auto* child_block : it->blocks()) { in PrepareForRemoveMutations() local
/external/pytorch/torch/csrc/jit/passes/
Dremove_inplace_ops.cpp104 for (auto* child_block : it->blocks()) { in ImplicitCastForBinaryInplaceOps() local