Searched refs:MapAsArrayWithLastDimAsRows (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | gru_cell.cc | 29 using optimized_ops::MapAsArrayWithLastDimAsRows;
68 auto ru = MapAsArrayWithLastDimAsRows(activation, activation_shape); in GruCell()
74 auto h = MapAsArrayWithLastDimAsRows(input_state, state_shape); in GruCell()
75 auto xh = MapAsArrayWithLastDimAsRows(concat, concat_shape); in GruCell()
84 auto c = MapAsArrayWithLastDimAsRows(output, output_shape); in GruCell()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops_utils.h | 80 ArrayMap<Scalar> MapAsArrayWithLastDimAsRows(Scalar* data, in MapAsArrayWithLastDimAsRows() function
|
| H A D | legacy_optimized_ops.h | 3069 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
3079 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
3081 MapAsArrayWithLastDimAsRows(output_state_data, output_state_shape); in LstmCell()
3083 MapAsArrayWithLastDimAsRows(output_activ_data, output_activ_shape); in LstmCell()
|
| H A D | optimized_ops.h | 2543 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
2553 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
2555 MapAsArrayWithLastDimAsRows(output_state_data, output_state_shape); in LstmCell()
2557 MapAsArrayWithLastDimAsRows(output_activ_data, output_activ_shape); in LstmCell()
|