Home
last modified time | relevance | path

Searched refs:concat_input_arrays_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h506 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
509 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
510 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
514 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
721 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
726 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3044 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
3047 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
3048 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
3052 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
3196 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
3201 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
Doptimized_ops.h2843 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
2846 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
2847 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
2851 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
2974 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
2979 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()