| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1860 const RuntimeShape prev_state_shape = in LstmCell() local
1874 MatchingDim(input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
1877 MatchingDim(input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
1880 MatchingDim(input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
1894 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
1942 prev_state_data[Offset(prev_state_shape, b, h, w, c)]; in LstmCell()
2072 const RuntimeShape prev_state_shape = in LstmCell() local
2087 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
2101 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3670 const RuntimeShape prev_state_shape = in LstmCell() local
3684 input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
3687 input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
3690 input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
3704 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
3743 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
3826 const RuntimeShape prev_state_shape = in LstmCell() local
3841 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
3855 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3579 const RuntimeShape prev_state_shape = in LstmCell() local
3593 input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
3596 input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
3599 input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
3613 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
3652 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
3710 const RuntimeShape prev_state_shape = in LstmCell() local
3725 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
3739 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1035 const auto& prev_state_shape = prev_state_array.shape(); in ProcessLstmCellOperator() local
1036 CHECK_GE(prev_state_shape.dimensions_count(), 2); in ProcessLstmCellOperator()
|