Home
last modified time | relevance | path

Searched refs:total_input_depth (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h488 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
490 total_input_depth); in LstmCell()
495 intern_activ_depth * total_input_depth); in LstmCell()
697 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
699 total_input_depth); in LstmCell()
703 intern_activ_depth * total_input_depth); in LstmCell()
713 const int fc_accum_depth = total_input_depth; in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3029 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
3031 total_input_depth); in LstmCell()
3036 intern_activ_depth * total_input_depth); in LstmCell()
3175 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
3177 total_input_depth); in LstmCell()
3181 intern_activ_depth * total_input_depth); in LstmCell()
3191 const int fc_accum_depth = total_input_depth; in LstmCell()
Doptimized_ops.h2825 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
2827 total_input_depth); in LstmCell()
2832 intern_activ_depth * total_input_depth); in LstmCell()
2950 const int total_input_depth = prev_activ_depth + input_depth; in LstmCell() local
2952 total_input_depth); in LstmCell()
2956 intern_activ_depth * total_input_depth); in LstmCell()
2966 const int fc_accum_depth = total_input_depth; in LstmCell()