Home
last modified time | relevance | path

Searched refs:flatten_perdevice_values (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py104 all_inputs = flatten_perdevice_values(distribution_strategy,
110 all_outputs = flatten_perdevice_values(distribution_strategy,
114 all_outputs = flatten_perdevice_values(distribution_strategy,
118 all_updates = flatten_perdevice_values(distribution_strategy,
127 all_session_args['feed_dict'] = flatten_perdevice_values(
132 all_session_args['fetches'] = flatten_perdevice_values(
139 def flatten_perdevice_values(distribution_strategy, perdevice_values): function
568 inputs = flatten_perdevice_values(strategy, inputs)
569 targets = flatten_perdevice_values(strategy, targets)