Searched refs:concat_temp_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1055 Shape concat_temp_shape(input_shape); in ProcessLstmCellOperator() local
1056 (*concat_temp_shape in ProcessLstmCellOperator()
1057 .mutable_dims())[concat_temp_shape.dimensions_count() - 1] = in ProcessLstmCellOperator()
1060 .copy_shape(concat_temp_shape); in ProcessLstmCellOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1866 const RuntimeShape concat_temp_shape = in LstmCell() local
1909 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
1916 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
2078 const RuntimeShape concat_temp_shape = in LstmCell() local
2119 concat_input_arrays_data, concat_temp_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3585 const RuntimeShape concat_temp_shape = in LstmCell() local
3628 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3635 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
3716 const RuntimeShape concat_temp_shape = in LstmCell() local
3757 concat_input_arrays_data, concat_temp_shape, in LstmCell()
3768 GEMVForLstmCell(concat_temp_shape, concat_temp_data_uint8, weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3676 const RuntimeShape concat_temp_shape = in LstmCell() local
3719 &(concat_input_arrays_data[0]), concat_temp_shape, in LstmCell()
3726 FullyConnected(fc_params, concat_temp_shape, concat_temp_data, weights_shape, in LstmCell()
3832 const RuntimeShape concat_temp_shape = in LstmCell() local
3873 concat_input_arrays_data, concat_temp_shape, in LstmCell()
3884 GEMVForLstmCell(concat_temp_shape, concat_temp_data_uint8, weights_shape, in LstmCell()
|