Colocate the Switch nodes for inputs to tf.cond with the input node in the
lowering pass. Enable control_flow_ops_py_test to run on GPU. PiperOrigin-RevId: 238580870
Loading
Please sign in to comment
lowering pass. Enable control_flow_ops_py_test to run on GPU. PiperOrigin-RevId: 238580870