Searched defs:loss_grad_layout (Results 1 – 1 of 1) sorted by relevance
2347 void SplitSens(const CNodePtr &grad_sens_node, const TensorLayout &loss_grad_layout) { in SplitSens()2482 auto loss_grad_layout = GetLossNodeGradOutputLayout(loss_node); in StepSplitSens() local