Searched refs:NumDistributedArgs (Results 1 – 2 of 2) sorted by relevance
1738 for (int64 i = 0; i < params_info.NumDistributedArgs(); ++i) { in GetArgAndRetvalShapes()1743 i < params_info.NumPerReplicaArgs() + params_info.NumDistributedArgs(); in GetArgAndRetvalShapes()1757 params_info.NumDistributedArgs()] in GetArgAndRetvalShapes()1770 params_info.NumDistributedArgs() + in GetArgAndRetvalShapes()1789 params_info.NumDistributedArgs() + in GetArgAndRetvalShapes()2247 params_info.NumDistributedArgs() + in BuildDynamicShapeNodes()2275 params_info.NumDistributedArgs() - in BuildDynamicShapeNodes()3077 params_info.NumDistributedArgs() - in BuildExecuteNodes()3195 params_info.NumDistributedArgs(); in BuildExecuteNodes()3209 params_info.NumDistributedArgs() - in BuildExecuteNodes()[all …]
163 int64 NumDistributedArgs() const { return num_distributed_args_; } in NumDistributedArgs() function