Searched refs:activ_temp_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1868 const RuntimeShape activ_temp_shape = in LstmCell() local
1917 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
1927 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
1930 activ_temp_shape, b, h, w, 1 * output_depth + c)]); in LstmCell()
1933 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
1937 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
2080 const RuntimeShape activ_temp_shape = in LstmCell() local
2104 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
2106 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1062 Shape activ_temp_shape(input_shape); in ProcessLstmCellOperator() local
1063 (*activ_temp_shape.mutable_dims())[activ_temp_shape.dimensions_count() - 1] = in ProcessLstmCellOperator()
1066 .copy_shape(activ_temp_shape); in ProcessLstmCellOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3587 const RuntimeShape activ_temp_shape = in LstmCell() local
3636 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
3642 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
3718 const RuntimeShape activ_temp_shape = in LstmCell() local
3742 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
3744 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
3771 activ_temp_shape, activ_temp_data_int16); in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3678 const RuntimeShape activ_temp_shape = in LstmCell() local
3727 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
3733 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
3834 const RuntimeShape activ_temp_shape = in LstmCell() local
3858 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
3860 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
3887 activ_temp_shape, activ_temp_data_int16); in LstmCell()
|