Searched refs:NumInputsToEachReplica (Results 1 – 2 of 2) sorted by relevance
209 int64 NumInputsToEachReplica() const { in NumInputsToEachReplica() function
1703 params_info.NumInputsToEachReplica()); in GetArgAndRetvalShapes()1705 arg_shapes->resize(params_info.NumInputsToEachReplica()); in GetArgAndRetvalShapes()4016 if (arg_types->size() != params_info->NumInputsToEachReplica()) { in GetIOTypes()4020 params_info->NumInputsToEachReplica(), " inputs, got ", in GetIOTypes()