Searched defs:output_state_dims (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 65 const Dims<4>& output_state_dims, uint8_t* output_activ_data_uint8, in quantizedLstmStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | legacy_reference_ops.h | 577 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell()
601 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3103 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell()
3414 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell()
|