Searched refs:fc_output_depth (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 711 const int fc_output_depth = in LstmCell() local
714 TFLITE_DCHECK_EQ(fc_output_depth, 4 * output_depth); in LstmCell()
734 for (int out_c = 0; out_c < fc_output_depth; ++out_c) { in LstmCell()
753 activ_temp_data_int16[out_c + fc_output_depth * b] = accum; in LstmCell()
781 activ_temp_data_int16[b * fc_output_depth + 0 * output_depth + c]); in LstmCell()
786 activ_temp_data_int16[b * fc_output_depth + 1 * output_depth + c]); in LstmCell()
791 activ_temp_data_int16[b * fc_output_depth + 2 * output_depth + c]); in LstmCell()
795 activ_temp_data_int16[b * fc_output_depth + 3 * output_depth + c]); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1051 const int fc_output_depth = weights_shape.dims(0); in ProcessLstmCellOperator() local
1052 CHECK_EQ(fc_output_depth, bias_shape.dims(0)); in ProcessLstmCellOperator()
1053 CHECK_EQ(fc_output_depth % 4, 0); in ProcessLstmCellOperator()
1054 const int depth = fc_output_depth / 4; in ProcessLstmCellOperator()
1077 fc_output_depth; in ProcessLstmCellOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3189 const int fc_output_depth = in LstmCell() local
3192 TFLITE_DCHECK_EQ(fc_output_depth, 4 * output_depth); in LstmCell()
3213 if (fc_batches == 1 && !(fc_output_depth % 4) && !(fc_accum_depth % 8)) { in LstmCell()
3223 weights_matrix(weights_data_uint8, fc_output_depth, fc_accum_depth); in LstmCell()
3227 activ_temp_data_int16, fc_output_depth, fc_batches); in LstmCell()
3230 ColVectorMap bias_vector(bias_data_int32, fc_output_depth); in LstmCell()
|
| D | optimized_ops.h | 2964 const int fc_output_depth = in LstmCell() local
2967 TFLITE_DCHECK_EQ(fc_output_depth, 4 * output_depth); in LstmCell()
2987 lhs_params.rows = fc_output_depth; in LstmCell()
2997 dst_params.rows = fc_output_depth; in LstmCell()
|