Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h2107 const int fc_accum_depth = total_input_depth; in LstmCell() local
2133 for (int d = 0; d < fc_accum_depth; ++d) { in LstmCell()
2134 int16 input_val = concat_temp_data_uint8[b * fc_accum_depth + d] - 128; in LstmCell()
2136 weights_data_uint8[out_c * fc_accum_depth + d] - weights_zero_point; in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3745 const int fc_accum_depth = total_input_depth; in LstmCell() local
3767 if (fc_batches == 1 && !(fc_output_depth % 4) && !(fc_accum_depth % 8)) { in LstmCell()
3777 weights_matrix(weights_data_uint8, fc_output_depth, fc_accum_depth); in LstmCell()
3779 concat_temp_data_uint8, fc_accum_depth, fc_batches); in LstmCell()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h3861 const int fc_accum_depth = total_input_depth; in LstmCell() local
3883 if (fc_batches == 1 && !(fc_output_depth % 4) && !(fc_accum_depth % 8)) { in LstmCell()
3893 weights_matrix(weights_data_uint8, fc_output_depth, fc_accum_depth); in LstmCell()
3895 concat_temp_data_uint8, fc_accum_depth, fc_batches); in LstmCell()