Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h185 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
187 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
189 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected()
Dreference_ops.h476 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
489 TFLITE_DCHECK_EQ(weights_shape.Dims(weights_dim_count - 1), in LstmCell()
493 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
691 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
698 TFLITE_DCHECK_EQ(weights_shape.Dims(weights_dim_count - 1), in LstmCell()
701 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
712 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1167 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCell() local
1170 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCell()
1354 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCellWithSymmetricRange() local
1357 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCellWithSymmetricRange()
2237 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
2239 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
2241 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected()
3017 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
3030 TFLITE_DCHECK_EQ(weights_shape.Dims(weights_dim_count - 1), in LstmCell()
3034 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
[all …]
Doptimized_ops.h834 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
836 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
838 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected()
2813 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
2826 TFLITE_DCHECK_EQ(weights_shape.Dims(weights_dim_count - 1), in LstmCell()
2830 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
2944 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
2951 TFLITE_DCHECK_EQ(weights_shape.Dims(weights_dim_count - 1), in LstmCell()
2954 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
2965 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()