Searched refs:concat_input_arrays_data (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 94 float const* concat_input_arrays_data[2] = {input_data, prev_activ_data}; in LstmCell() local
101 concat_input_arrays_data, concat_temp_shape, concat_temp_data); in LstmCell()
304 uint8_t const* concat_input_arrays_data[2] = {input_data_uint8, in LstmCell() local
312 concat_input_arrays_data, concat_temp_shape, in LstmCell()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 96 uint8_t const* concat_input_arrays_data[2] = {input_data_uint8, in quantizedLstmStep() local
100 0, concat_input_arrays_data, concat_input_arrays_dims, 2, in quantizedLstmStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3045 std::vector<float const*> concat_input_arrays_data; in LstmCell() local
3047 concat_input_arrays_data.push_back(input_data); in LstmCell()
3048 concat_input_arrays_data.push_back(prev_activ_data); in LstmCell()
3053 concat_params.inputs_count = concat_input_arrays_data.size(); in LstmCell()
3055 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3196 uint8 const* concat_input_arrays_data[2] = {input_data_uint8, in LstmCell() local
3204 concat_input_arrays_data, concat_temp_shape, in LstmCell()
|
| H A D | optimized_ops.h | 2517 std::vector<float const*> concat_input_arrays_data; in LstmCell() local
2519 concat_input_arrays_data.push_back(input_data); in LstmCell()
2520 concat_input_arrays_data.push_back(prev_activ_data); in LstmCell()
2525 concat_params.inputs_count = concat_input_arrays_data.size(); in LstmCell()
2527 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
2647 uint8 const* concat_input_arrays_data[2] = {input_data_uint8, in LstmCell() local
2655 concat_input_arrays_data, concat_temp_shape, in LstmCell()
|