Searched refs:multi_gpu_model (Results 1 – 3 of 3) sorted by relevance
76 parallel_model = multi_gpu_utils.multi_gpu_model(model, gpus=gpus)79 parallel_model = multi_gpu_utils.multi_gpu_model(113 parallel_model = multi_gpu_utils.multi_gpu_model(model, gpus=gpus)117 parallel_model = multi_gpu_utils.multi_gpu_model(138 parallel_model = multi_gpu_utils.multi_gpu_model(143 parallel_model = multi_gpu_utils.multi_gpu_model(148 parallel_model = multi_gpu_utils.multi_gpu_model(model, gpus=1)152 parallel_model = multi_gpu_utils.multi_gpu_model(model, gpus=[0])191 parallel_model = multi_gpu_utils.multi_gpu_model(outer_model, gpus=gpus)214 parallel_model = multi_gpu_utils.multi_gpu_model(model, gpus=gpus)[all …]
38 from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
37 def multi_gpu_model(model, gpus, cpu_merge=True, cpu_relocation=False): function