Searched refs:loss_node (Results 1 – 2 of 2) sorted by relevance
47 CNodePtr loss_node = nullptr; member
1231 …e](const std::pair<CNodePtr, LossNodeInfo> &element) { return element.second.loss_node == node; }); in BackwardCommunication()2278 loss_node_info.loss_node = pre_cnode; in FindLossCNode()2296 loss_node_info.loss_node = pre_pre_cnode; in FindLossCNode()2307 loss_node_info.loss_node = pre_cnode; in FindLossCNode()2314 auto loss_cnode = node_info.loss_node; in GetLossNodeGradOutputLayout()2481 auto loss_node = sens_loss_pair.second; in StepSplitSens() local2482 auto loss_grad_layout = GetLossNodeGradOutputLayout(loss_node); in StepSplitSens()2534 if (loss_node_info.loss_node == nullptr) { in GetSensLossPairs()2760 auto loss_cnode = FindLossCNode(graph, 0).loss_node; in FindRootForwardCNode()