Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/
H A Dpattern_conversion.cpp70 const std::vector<Node*>& slice_and_select_nodes, in MergeSliceAndSelectToIndices()
305 std::vector<Node*> slice_and_select_nodes = in ConvertIndexPutToONNX() local
H A Dpattern_encapsulation.cpp34 std::vector<Node*> slice_and_select_nodes = in EncapsulateInplaceIndexPutForONNX() local