Searched defs:time_steps (Results 1 – 2 of 2) sorted by relevance
172 int time_steps = static_cast<int>(GetNumberOfSteps(input) / num_features); in FlexibleLayerNorm() local289 const int time_steps = in IntegerLayerNorm() local332 const int time_steps = in DefaultLayerNormFloat() local360 const int time_steps = in DefaultLayerNorm() local
3175 time_steps=None, argument