Searched refs:get_input_params (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_strategy_test.py | 380 steps, batch_size = distributed_training_utils_v1.get_input_params(
386 steps, batch_size = distributed_training_utils_v1.get_input_params(
402 steps, batch_size = distributed_training_utils_v1.get_input_params(
408 steps, batch_size = distributed_training_utils_v1.get_input_params(
415 distributed_training_utils_v1.get_input_params(
422 steps, batch_size = distributed_training_utils_v1.get_input_params(
431 distributed_training_utils_v1.get_input_params(
445 steps, batch_size = distributed_training_utils_v1.get_input_params(
451 steps, batch_size = distributed_training_utils_v1.get_input_params(
461 steps, batch_size = distributed_training_utils_v1.get_input_params(
[all …]
|
| D | distributed_training_utils_v1.py | 453 steps_per_epoch, batch_size = get_input_params(
458 def get_input_params(distribution_strategy, function
|