Searched refs:enable_xla_param_broadcast_ (Results 1 – 2 of 2) sorted by relevance
587 static bool enable_xla_param_broadcast_; variable
2338 enable_xla_param_broadcast_); in BuildCompileNode()3245 arg_shardings, replicate_node, enable_xla_param_broadcast_, in BuildExecuteNodes()4540 bool DistributedTPURewritePass::enable_xla_param_broadcast_ = false; member in tensorflow::DistributedTPURewritePass4554 enable_xla_param_broadcast_ = enable_xla_param_broadcast; in SetDistributedTpuRewritePassOptions()