Lines Matching refs:_config
58 self._config = config
93 config = self._config
110 run_config=self._config,
127 computation_shape=self._config.tpu_config.computation_shape,
131 str(self._config.tpu_config.computation_shape))
173 self._config.tpu_config.computation_shape, dtype=np.int32)
181 self._config.tpu_config.computation_shape,
202 return self._config
208 not self._config.tpu_config.per_host_input_for_training)
304 run_config = self._config
414 if self._config.tpu_config.num_shards:
415 user_provided_num_replicas = self._config.tpu_config.num_shards
