Searched refs:_build_distributed_network (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distributed_training_utils_v1.py | 754 def _build_distributed_network(model, strategy, mode, inputs=None, function
891 _build_distributed_network(model, strategy, mode)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_distributed_v1.py | 51 dist_utils._build_distributed_network(model, strategy, mode, inputs,
|