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.h504 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
507 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
508 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
512 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
719 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
724 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3045 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
3048 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
3049 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
3053 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
3197 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
3202 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()
Doptimized_ops.h2841 std::vector<RuntimeShape const*> concat_input_arrays_shapes; in LstmCell() local
2844 concat_input_arrays_shapes.push_back(&input_shape); in LstmCell()
2845 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
2849 Concatenation(concat_params, &(concat_input_arrays_shapes[0]), in LstmCell()
2972 const RuntimeShape* concat_input_arrays_shapes[2] = {&input_shape, in LstmCell() local
2977 Concatenation(concat_params, concat_input_arrays_shapes, in LstmCell()