Searched refs:activ_temp_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1845 const RuntimeShape& unextended_activ_temp_shape, float* activ_temp_data) { in LstmCell() argument
1918 activ_temp_data); in LstmCell()
1927 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
1929 const float new_input = std::tanh(activ_temp_data[Offset( in LstmCell()
1933 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
1937 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
|
| D | legacy_reference_ops.h | 527 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
539 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 828 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
840 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
|
| D | optimized_ops.h | 3563 const RuntimeShape& unextended_activ_temp_shape, float* activ_temp_data) { in LstmCell() argument
3637 activ_temp_data); in LstmCell()
3642 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3654 const RuntimeShape& unextended_activ_temp_shape, float* activ_temp_data) { in LstmCell() argument
3728 activ_temp_data); in LstmCell()
3733 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
3771 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
3783 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
|