Searched refs:first_x_value (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 64 first_x_value = nest.flatten(x)[0]
65 if isinstance(first_x_value, np.ndarray):
71 model._distribution_strategy, first_x_value, steps_per_epoch,
145 first_x_value = nest.flatten(x)[0]
146 if isinstance(first_x_value, np.ndarray):
148 model._distribution_strategy, first_x_value, steps, batch_size)
177 first_x_value = nest.flatten(x)[0]
178 if isinstance(first_x_value, np.ndarray):
180 model._distribution_strategy, first_x_value, steps,
|
| D | distributed_training_utils.py | 406 def get_input_params(distribution_strategy, first_x_value, steps, batch_size, argument
431 num_samples = first_x_value.shape[0]
|
| D | training.py | 2237 first_x_value = nest.flatten(x)[0]
2238 if isinstance(first_x_value, np.ndarray):
|