Searched defs:enable_xla_param_broadcast (Results 1 – 1 of 1) sorted by relevance
2576 bool enable_xla_param_broadcast, bool mpmd, in EnableXlaParamBroadcast()3156 const Node& replicate_node, const bool enable_xla_param_broadcast, in CreateOrGetPerHostVariableCopy()5074 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast, in SetDistributedTpuRewritePassOptions()