Searched defs:mpmd (Results 1 – 1 of 1) sorted by relevance
2576 bool enable_xla_param_broadcast, bool mpmd, in EnableXlaParamBroadcast()2624 const bool mpmd = (num_cores_per_replica > 1) && !use_spmd; in BuildCompileNode() local3157 const bool mpmd, const int num_cores_per_replica, int replica_id, in CreateOrGetPerHostVariableCopy()3272 const bool mpmd = (num_cores_per_replica > 1) && !use_spmd; in BuildExecuteNodes() local