| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 464 const RuntimeShape prev_state_shape = in LstmCell() local
478 MatchingDim(input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
481 MatchingDim(input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
484 MatchingDim(input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
498 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
546 prev_state_data[Offset(prev_state_shape, b, h, w, c)]; in LstmCell()
678 const RuntimeShape prev_state_shape = in LstmCell() local
693 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
707 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3005 const RuntimeShape prev_state_shape = in LstmCell() local
3019 input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
3022 input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
3025 input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
3039 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
3078 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
3156 const RuntimeShape prev_state_shape = in LstmCell() local
3171 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
3185 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
|
| D | optimized_ops.h | 2801 const RuntimeShape prev_state_shape = in LstmCell() local
2815 input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
2818 input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
2821 input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
2835 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
2876 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
2931 const RuntimeShape prev_state_shape = in LstmCell() local
2946 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
2960 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 | 1048 const auto& prev_state_shape = prev_state_array.shape(); in ProcessLstmCellOperator() local
1049 CHECK_GE(prev_state_shape.dimensions_count(), 2); in ProcessLstmCellOperator()
|