Searched refs:BroadCastMicroBatch (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.h | 59 void BroadCastMicroBatch(const CNodePtr &node, NodeUsersMap *node_users_map, const ValuePtr &value,…
|
| D | pipeline_split_utils.cc | 413 void BroadCastMicroBatch(const CNodePtr &node, NodeUsersMap *node_users_map, const ValuePtr &value,… in BroadCastMicroBatch() function
424 BroadCastMicroBatch(user_node, node_users_map, value, max_depth + 1); in BroadCastMicroBatch()
568 BroadCastMicroBatch(cnode, &node_users_map, micro, 0); in HandleMicroBatch()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 172 BroadCastMicroBatch(cnode, &node_user_map, micro, 0); in LabelMicroBatch()
|