Searched refs:concat_input_arrays_shapes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 95 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
100 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
306 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
311 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3046 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
3049 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
3050 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
3054 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
3198 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
3203 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
|
| H A D | optimized_ops.h | 2518 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
2521 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
2522 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
2526 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
2649 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
2654 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
|