Searched refs:multi_gpu_model (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/keras/_impl/keras/utils/
H A Dtraining_utils_test.py47 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
51 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
81 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
85 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
102 parallel_model = keras.utils.multi_gpu_model(
107 parallel_model = keras.utils.multi_gpu_model(
112 parallel_model = keras.utils.multi_gpu_model(model, gpus=1)
116 parallel_model = keras.utils.multi_gpu_model(model, gpus=[0])
H A D__init__.py36 from tensorflow.python.keras._impl.keras.utils.training_utils import multi_gpu_model namespace
H A Dtraining_utils.py36 @tf_export('keras.utils.multi_gpu_model')
37 def multi_gpu_model(model, gpus): function
74 from keras.utils import multi_gpu_model
94 parallel_model = multi_gpu_model(model, gpus=8)
120 'call `multi_gpu_model` with `len(gpus) >= 2`. '
127 'call `multi_gpu_model` with `gpus >= 2`. '
139 raise ValueError('To call `multi_gpu_model` with `gpus=%d`, '
/external/tensorflow/tensorflow/python/keras/utils/
H A D__init__.py35 from tensorflow.python.keras._impl.keras.utils.training_utils import multi_gpu_model namespace

Completed in 156 milliseconds