Searched refs:GetLossNodeGradOutputLayout (Results 1 – 1 of 1) sorted by relevance
2312 TensorLayouts GetLossNodeGradOutputLayout(const LossNodeInfo &node_info) { in GetLossNodeGradOutputLayout() function2482 auto loss_grad_layout = GetLossNodeGradOutputLayout(loss_node); in StepSplitSens()