Home
last modified time | relevance | path

Searched defs:dist_model (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_optimizers.py56 dist_model, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/
H A Ddistributed_training_utils_v1.py51 def set_weights(distribution_strategy, dist_model, weights): argument