Searched refs:output_activ_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 826 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
838 DimsToShape(output_activ_dims), output_activ_data, in LstmCell()
|
| D | optimized_ops.h | 3561 const RuntimeShape& unextended_output_activ_shape, float* output_activ_data, in LstmCell() argument
3656 MapAsArrayWithLastDimAsRows(output_activ_data, output_activ_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 525 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
537 DimsToShape(output_activ_dims), output_activ_data, in LstmCell()
|
| D | reference_ops.h | 1843 const RuntimeShape& unextended_output_activ_shape, float* output_activ_data, in LstmCell() argument
1944 output_activ_data[Offset(output_activ_shape, b, h, w, c)] = in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3652 const RuntimeShape& unextended_output_activ_shape, float* output_activ_data, in LstmCell() argument
3747 MapAsArrayWithLastDimAsRows(output_activ_data, output_activ_shape); in LstmCell()
3769 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
3781 DimsToShape(output_activ_dims), output_activ_data, in LstmCell()
|