Searched refs:IsPerReplicaArg (Results 1 – 2 of 2) sorted by relevance
1996 const bool is_per_replica_arg = params_info.IsPerReplicaArg(i); in AssignArgsAndRetvalsToCores()2042 ((params_info.IsPerReplicaArg(i) || in AssignArgsAndRetvalsToCores()2054 if (((params_info.IsPerReplicaArg(i) || in AssignArgsAndRetvalsToCores()2258 if (params_info.IsPerReplicaArg(i)) { in BuildDynamicShapeNodes()2391 !params_info.IsPerReplicaArg(i) && !params_info.IsDistributedArg(i) && in BuildCompileNode()3071 !params_info.IsPerReplicaArg(input) && in BuildExecuteNodes()3118 if (params_info.IsPerReplicaArg(orig_arg_num) || in BuildExecuteNodes()3121 int64 input_num = params_info.IsPerReplicaArg(orig_arg_num) in BuildExecuteNodes()4035 TF_RET_CHECK(params_info->IsPerReplicaArg(index) || in GetIOTypes()
173 bool IsPerReplicaArg(int64 index) const { in IsPerReplicaArg() function