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() local236 const int time_steps = in DefaultLayerNormFloat() local264 const int time_steps = in DefaultLayerNorm() local
3099 time_steps=None, argument