Searched refs:enable_xla_param_broadcast (Results 1 – 2 of 2) sorted by relevance
135 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast);
2794 const Node& replicate_node, const bool enable_xla_param_broadcast, in CreateOrGetPerHostVariableCopy() argument2846 if (parsed_device.task == 0 || !enable_xla_param_broadcast) { in CreateOrGetPerHostVariableCopy()4546 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast) { in SetDistributedTpuRewritePassOptions() argument4554 enable_xla_param_broadcast_ = enable_xla_param_broadcast; in SetDistributedTpuRewritePassOptions()