Searched defs:num_time_steps (Results 1 – 3 of 3) sorted by relevance
18 def __call__(self, prompt, height, width, num_time_steps): argument
122 auto num_time_steps = inputs.size(); in CalculateLoss() local
37 const int num_time_steps, in Runner()