Step 1 toward automatically assigning # of TPU cores.
Also fix the API as num_cores is not a good API. Changed to a debugging oriented argument using_single_core. Ideally we should not need that in future. The updated lstm example cannot work with TF 1.10 anymore. PiperOrigin-RevId: 210632592
Loading
Please sign in to comment