Home
last modified time | relevance | path

Searched refs:activ_temp_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h472 const RuntimeShape activ_temp_shape = in LstmCell() local
521 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
531 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
534 activ_temp_shape, b, h, w, 1 * output_depth + c)]); in LstmCell()
537 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
541 (1.f + std::exp(-activ_temp_data[Offset(activ_temp_shape, b, h, w, in LstmCell()
686 const RuntimeShape activ_temp_shape = in LstmCell() local
710 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
712 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1075 Shape activ_temp_shape(input_shape); in ProcessLstmCellOperator() local
1076 (*activ_temp_shape.mutable_dims())[activ_temp_shape.dimensions_count() - 1] = in ProcessLstmCellOperator()
1079 .copy_shape(activ_temp_shape); in ProcessLstmCellOperator()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3013 const RuntimeShape activ_temp_shape = in LstmCell() local
3062 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
3068 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
3164 const RuntimeShape activ_temp_shape = in LstmCell() local
3188 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
3190 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
3217 activ_temp_shape, activ_temp_data_int16); in LstmCell()
Doptimized_ops.h2809 const RuntimeShape activ_temp_shape = in LstmCell() local
2860 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
2866 MapAsArrayWithLastDimAsRows(activ_temp_data, activ_temp_shape); in LstmCell()
2939 const RuntimeShape activ_temp_shape = in LstmCell() local
2963 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
2965 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()