Home
last modified time | relevance | path

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

/aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
H A Dlayer_norm.cc172 int time_steps = static_cast<int>(GetNumberOfSteps(input) / num_features); in FlexibleLayerNorm() local
289 const int time_steps = in IntegerLayerNorm() local
332 const int time_steps = in DefaultLayerNormFloat() local
360 const int time_steps = in DefaultLayerNorm() local
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/stable_diffusion/runner/
H A Drunner.cpp249 std::vector<float> time_steps(num_time_steps_); in get_time_steps() local
257 std::vector<float> Runner::get_sigmas(const std::vector<float>& time_steps) { in get_sigmas()
394 std::vector<float> time_steps = get_time_steps(); in generate() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Drnn_cell_test.py3175 time_steps=None, argument