Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dsqueeze_fusion.cc75 auto squeeze_node = bn_cnode->input(1); in Process() local
Dtflite_lstm_cell_fusion.cc679 auto squeeze_node = CreateSqueezeNode(func_graph, get_item, squeeze_axis); in AdjustOtherGetItems() local
803 auto squeeze_node = CreateSqueezeNode(func_graph, get_item_node, squeeze_axis); in Process() local