Searched refs:op_output_size (Results 1 – 1 of 1) sorted by relevance
2329 size_t op_output_size = operator_info->outputs_tensor_info().size(); in GetLossNodeGradOutputLayout() local2331 …<< node_info.has_tuple_getitem << ", the output size is " << op_output_size << ", the dout_index … in GetLossNodeGradOutputLayout()2334 if ((op_output_size == 0) || (op_output_size <= LongToSize(node_info.dout_index))) { in GetLossNodeGradOutputLayout()2335 … << "The index is " << node_info.dout_index << ", but the size of outputs is " << op_output_size; in GetLossNodeGradOutputLayout()2338 if (!node_info.has_tuple_getitem && (op_output_size > 1)) { in GetLossNodeGradOutputLayout()