Home
last modified time | relevance | path

Searched defs:index_put_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/
H A Dpattern_conversion.cpp69 Node* index_put_node, in MergeSliceAndSelectToIndices()
189 Node* index_put_node, in ReshapeToAdvancedIndexingFormat()
300 auto index_put_node = subblock->nodes().back()->prev(); in ConvertIndexPutToONNX() local
H A Dpattern_encapsulation.cpp28 Node* EncapsulateInplaceIndexPutForONNX(Node* index_put_node) { in EncapsulateInplaceIndexPutForONNX()