Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Dspliter.h57 std::unordered_map<std::string, std::vector<ShapeVector>> graph_node_input_shapes() const { in graph_node_input_shapes() function
Dmulti_conv_info.cc189 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in SplitSingleConv() local
Dconv2d_info.cc151 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in CreateOutputsOfSplit() local
Ddepthwise_conv2d_info.cc348 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in CreateOutputsOfSplit() local