Home
last modified time | relevance | path

Searched defs:total_input_depth (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlstm_cell.h79 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
284 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTM.cpp78 const int total_input_depth = prev_activ_depth + input_depth; in quantizedLstmStep() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h3030 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
3176 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
H A Doptimized_ops.h2502 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
2627 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local