Searched defs:output_state_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 58 float* output_state_data = GetTensorData<float>(output_state); in GruImpl() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 446 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|
| D | legacy_reference_ops.h | 574 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 2986 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
3101 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
|
| D | optimized_ops.h | 2781 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|