Home
last modified time | relevance | path

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

/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/reference/
Dreference_ops.h1151 const RuntimeShape concat_temp_shape = in LstmCell() local
1194 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
1201 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
1365 const RuntimeShape concat_temp_shape = in LstmCell() local
1406 concat_input_arrays_data, concat_temp_shape, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2998 const RuntimeShape concat_temp_shape = in LstmCell() local
3041 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3048 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
3149 const RuntimeShape concat_temp_shape = in LstmCell() local
3190 concat_input_arrays_data, concat_temp_shape, in LstmCell()
3201 GEMVForLstmCell(concat_temp_shape, concat_temp_data_uint8, weights_shape, in LstmCell()
Doptimized_ops.h2964 const RuntimeShape concat_temp_shape = in LstmCell() local
3007 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3016 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
3094 const RuntimeShape concat_temp_shape = in LstmCell() local
3135 concat_input_arrays_data, concat_temp_shape, in LstmCell()