Searched refs:IsDistributedArg (Results 1 – 2 of 2) sorted by relevance
177 bool IsDistributedArg(int64 index) const { in IsDistributedArg() function
1997 if (is_per_replica_arg || params_info.IsDistributedArg(i)) { in AssignArgsAndRetvalsToCores()2043 params_info.IsDistributedArg(i)) && in AssignArgsAndRetvalsToCores()2055 params_info.IsDistributedArg(i)) && in AssignArgsAndRetvalsToCores()2264 } else if (params_info.IsDistributedArg(i) || in BuildDynamicShapeNodes()2391 !params_info.IsPerReplicaArg(i) && !params_info.IsDistributedArg(i) && in BuildCompileNode()3072 !params_info.IsDistributedArg(input)) { in BuildExecuteNodes()3119 params_info.IsDistributedArg(orig_arg_num)) { in BuildExecuteNodes()4036 params_info->IsDistributedArg(index)) in GetIOTypes()