Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/
H A Dpattern_encapsulation.h31 TORCH_API std::optional<Node*> EncapsulatePatternIntoSubblock(Node* n);
H A Dpattern_encapsulation.cpp80 std::optional<Node*> EncapsulatePatternIntoSubblock(Node* n) { in EncapsulatePatternIntoSubblock() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dremove_inplace_ops_for_onnx.cpp169 auto placeholder_node = EncapsulatePatternIntoSubblock(node).value(); in PrepareIndexPutForONNX()