Home
last modified time | relevance | path

Searched defs:warmup_steps (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/model_averaging/
H A Dhierarchical_model_averager.py101 def __init__(self, period_group_size_dict=None, warmup_steps=0, process_group=None): argument
H A Daveragers.py88 def __init__(self, period, warmup_steps=0, process_group=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_with_tracking_alloc_test.cc142 int warmup_steps = 10; in TEST() local