Searched refs:virtual_div_op (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | softmax_entropy_loss_info_test.cc | 163 OperatorVector virtual_div_op = loss->virtual_div_op(); in TEST_F() local
165 OperatorArgs operator_args = virtual_div_op.at(0).second; in TEST_F()
171 ASSERT_EQ(virtual_div_op.at(0).first, "_VirtualDiv"); in TEST_F()
172 ASSERT_EQ(virtual_div_op.size(), 1); in TEST_F()
|
| D | matmul_info_test.cc | 387 OperatorVector virtual_div_op = matmul1->virtual_div_op(); in TEST_F() local
389 OperatorArgs operator_args = virtual_div_op.at(0).second; in TEST_F()
395 ASSERT_EQ(virtual_div_op.at(0).first, "_VirtualDiv"); in TEST_F()
396 ASSERT_EQ(virtual_div_op.size(), 1); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 83 void InsertVirtualDivOp(const VirtualDivOp &virtual_div_op, const CNodePtr &node);
|
| D | step_parallel.cc | 830 void InsertVirtualDivOp(const VirtualDivOp &virtual_div_op, const CNodePtr &node) { in InsertVirtualDivOp() argument
852 for (size_t pos = 0; pos < virtual_div_op.size(); ++pos) { in InsertVirtualDivOp()
854 InsertNode(virtual_div_op[pos], node, index, node->input(index), func_graph, instance_name); in InsertVirtualDivOp()
1234 VirtualDivOp virtual_div_op = distribute_operator->virtual_div_op(); in BackwardCommunication() local
1241 if (!virtual_div_op.empty() && is_loss_cnode && IsLastStage()) { in BackwardCommunication()
1243 InsertVirtualDivOp(virtual_div_op, node); in BackwardCommunication()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.h | 120 VirtualDivOp virtual_div_op() const { return virtual_div_op_; } in virtual_div_op() function
|