Searched refs:NumBroadcastArgs (Results 1 – 2 of 2) sorted by relevance
1752 for (int64 i = 0; i < params_info.NumBroadcastArgs(); ++i) { in GetArgAndRetvalShapes()1771 params_info.NumBroadcastArgs()]; in GetArgAndRetvalShapes()1790 params_info.NumBroadcastArgs() + params_info.NumVariables()] in GetArgAndRetvalShapes()2248 params_info.NumBroadcastArgs() + in BuildDynamicShapeNodes()2276 params_info.NumBroadcastArgs(); in BuildDynamicShapeNodes()2818 i + params_info.NumPerReplicaArgs() + params_info.NumBroadcastArgs(); in CreateOrGetPerHostVariableCopy()2857 params_info.NumBroadcastArgs(); in CreateOrGetPerHostVariableCopy()3078 params_info.NumBroadcastArgs(); in BuildExecuteNodes()3210 params_info.NumBroadcastArgs(); in BuildExecuteNodes()3385 params_info.NumDistributedArgs() + params_info.NumBroadcastArgs(); in BuildExecuteNodes()
165 int64 NumBroadcastArgs() const { return num_broadcast_args_; } in NumBroadcastArgs() function