Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h492 const int intern_activ_depth = in LstmCell() local
495 intern_activ_depth * total_input_depth); in LstmCell()
496 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
500 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()
700 const int intern_activ_depth = in LstmCell() local
703 intern_activ_depth * total_input_depth); in LstmCell()
705 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
709 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3033 const int intern_activ_depth = in LstmCell() local
3036 intern_activ_depth * total_input_depth); in LstmCell()
3037 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
3041 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()
3178 const int intern_activ_depth = in LstmCell() local
3181 intern_activ_depth * total_input_depth); in LstmCell()
3183 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
3187 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()
Doptimized_ops.h2829 const int intern_activ_depth = in LstmCell() local
2832 intern_activ_depth * total_input_depth); in LstmCell()
2833 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
2837 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()
2953 const int intern_activ_depth = in LstmCell() local
2956 intern_activ_depth * total_input_depth); in LstmCell()
2958 TFLITE_DCHECK_EQ(intern_activ_depth % 4, 0); in LstmCell()
2962 TFLITE_DCHECK_EQ(output_depth, intern_activ_depth / 4); in LstmCell()