Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dspliter.h59 std::unordered_map<std::string, std::vector<ShapeVector>> graph_node_input_shapes() const { in graph_node_input_shapes() function
Dconv2d_info.cc148 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in CreateOutputsOfSplit() local
Dmulti_conv_info.cc187 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in SplitSingleConv() local
Ddepthwise_conv2d_info.cc338 auto graph_node_input_shapes = Spliter::GetInstance()->graph_node_input_shapes(); in CreateOutputsOfSplit() local