Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dreshape_gather_reshape_fusion_pass.cc87 auto &gather_node = graph->nodes.at(gather_index); in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/optimizer/
Dtensorrt_optimizer.cc221 auto gather_node = resize_input1->cast<CNodePtr>(); in OptResizeHeightWidth() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dinline_control_flow_scheduler.cc653 const auto &gather_node = condition_gather_actor->kernel(); in HandleConditionGatherActor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1251 ops::Gather gather_node; in GenGatherNode() local
1280 ops::Gather gather_node; in GenGatherNodeDynamicIndex() local