Home
last modified time | relevance | path

Searched refs:concat_temp_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h470 const RuntimeShape concat_temp_shape = in LstmCell() local
513 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
520 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
684 const RuntimeShape concat_temp_shape = in LstmCell() local
725 concat_input_arrays_data, concat_temp_shape, in LstmCell()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1068 Shape concat_temp_shape(input_shape); in ProcessLstmCellOperator() local
1069 (*concat_temp_shape in ProcessLstmCellOperator()
1070 .mutable_dims())[concat_temp_shape.dimensions_count() - 1] = in ProcessLstmCellOperator()
1073 .copy_shape(concat_temp_shape); in ProcessLstmCellOperator()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3011 const RuntimeShape concat_temp_shape = in LstmCell() local
3054 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3061 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
3162 const RuntimeShape concat_temp_shape = in LstmCell() local
3203 concat_input_arrays_data, concat_temp_shape, in LstmCell()
3214 GEMVForLstmCell(concat_temp_shape, concat_temp_data_uint8, weights_shape, in LstmCell()
Doptimized_ops.h2807 const RuntimeShape concat_temp_shape = in LstmCell() local
2850 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
2859 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
2937 const RuntimeShape concat_temp_shape = in LstmCell() local
2978 concat_input_arrays_data, concat_temp_shape, in LstmCell()