def _configure(self, device_id: str) -> None:
    if ('GPU' in device_id):
        tf.config.set_visible_devices(device_id, 'GPU')
    self._config_proto = dict(allow_soft_placement=True, log_device_placement=False, intra_op_parallelism_threads=self._num_threads, inter_op_parallelism_threads=self._num_threads)
def _configure(self, device_id: str) -> None:
    if ('GPU' in device_id):
        tf.config.set_visible_devices(device_id, 'GPU')
    self._config_proto = dict(allow_soft_placement=True, log_device_placement=False, intra_op_parallelism_threads=self._num_threads, inter_op_parallelism_threads=self._num_threads)