Searched refs:prev_activ_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1856 const RuntimeShape prev_activ_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()
1883 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
1894 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
1904 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
2068 const RuntimeShape prev_activ_shape = in LstmCell() local
2087 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
2090 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3575 const RuntimeShape prev_activ_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()
3602 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
3613 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
3623 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
3706 const RuntimeShape prev_activ_shape = in LstmCell() local
3725 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
3728 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3666 const RuntimeShape prev_activ_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()
3693 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
3704 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
3714 concat_input_arrays_shapes.push_back(&prev_activ_shape); in LstmCell()
3822 const RuntimeShape prev_activ_shape = in LstmCell() local
3841 input_shape, 3, prev_activ_shape, prev_state_shape, output_state_shape, in LstmCell()
3844 const int prev_activ_depth = prev_activ_shape.Dims(3); in LstmCell()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1008 const auto& prev_activ_shape = prev_activ_array.shape(); in ProcessLstmCellOperator() local
1009 CHECK_GE(prev_activ_shape.dimensions_count(), 2); in ProcessLstmCellOperator()
|