Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/utils/
Dmulti_gpu_utils_test.py60 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
63 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
96 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
100 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
120 parallel_model = keras.utils.multi_gpu_model(
125 parallel_model = keras.utils.multi_gpu_model(
130 parallel_model = keras.utils.multi_gpu_model(model, gpus=1)
134 parallel_model = keras.utils.multi_gpu_model(model, gpus=[0])
173 parallel_model = keras.utils.multi_gpu_model(outer_model, gpus=gpus)
196 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
[all …]
D__init__.py39 from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
Dmulti_gpu_utils.py37 def multi_gpu_model(model, gpus, cpu_merge=True, cpu_relocation=False): function
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.utils.pbtxt56 name: "multi_gpu_model"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.utils.pbtxt56 name: "multi_gpu_model"