Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dpasses.cpp1275 auto* split_and_squeeze_node = graph->create( in UseSplitAndSqueeze() local
1279 split_and_squeeze_node->addInput(node->input(0)); in UseSplitAndSqueeze()
1280 split_and_squeeze_node->addInput(node->input(1)); in UseSplitAndSqueeze()
1281 split_and_squeeze_node->addInput(node->input(2)); in UseSplitAndSqueeze()
1282 split_and_squeeze_node->insertBefore(node); in UseSplitAndSqueeze()
1285 split_and_squeeze_node->output(i)->copyMetadata(squeeze_node->output()); in UseSplitAndSqueeze()
1287 split_and_squeeze_node->output(i)); in UseSplitAndSqueeze()