Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:_distribution_strategy (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/python/keras/engine/
training_arrays.py	138 input_iterator = _get_iterator(inputs, model._distribution_strategy) 144 if model._distribution_strategy: 146 strategy=model._distribution_strategy, 189 val_iterator = _get_iterator(val_inputs, model._distribution_strategy) 300 if model._distribution_strategy: 404 _reinitialize_iterator(val_iterator, model._distribution_strategy) 413 _reinitialize_iterator(input_iterator, model._distribution_strategy) 417 if model._distribution_strategy: 465 if model._distribution_strategy: 468 inputs, model._distribution_strategy)     [all...]
training_distributed.py	62 x, y, model._distribution_strategy) 71 model._distribution_strategy, first_x_value, steps_per_epoch, 89 val_x, val_y, model._distribution_strategy) 93 model._distribution_strategy, first_valx_value, validation_steps, 106 if distributed_training_utils.is_tpu_strategy(model._distribution_strategy): 144 distributed_training_utils.validate_inputs(x, y, model._distribution_strategy) 148 model._distribution_strategy, first_x_value, steps, batch_size) 155 if distributed_training_utils.is_tpu_strategy(model._distribution_strategy): 176 x, None, model._distribution_strategy, allow_partial_batch=True) 180 model._distribution_strategy, first_x_value, steps     [all...]
training.py	123 # initializing _distribution_strategy here since it is possible to call 125 self._distribution_strategy = None 139 if self._distribution_strategy: 140 with self._distribution_strategy.scope(): 146 if distributed_training_utils.is_tpu_strategy(self._distribution_strategy): 147 if (self._distribution_strategy.extended.steps_per_run > 1 and 234 self._distribution_strategy = distribute 243 self._distribution_strategy = ( 248 if self._distribution_strategy: 298 if self._distribution_strategy is not None     [all...]
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
keras_support.py	[all...]

Completed in 331 milliseconds