Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/
H A Dpattern_conversion.cpp73 std::unordered_map<int64_t, ConvertedIndex> dim_index_map; in MergeSliceAndSelectToIndices() local
190 std::unordered_map<int64_t, ConvertedIndex>& dim_index_map) { in ReshapeToAdvancedIndexingFormat()
315 std::unordered_map<int64_t, ConvertedIndex> dim_index_map = in ConvertIndexPutToONNX() local