Home
last modified time | relevance | path

Searched defs: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
Dreference_ops.h478 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
693 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1166 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCell() local
1353 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCellWithSymmetricRange() local
2236 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
3016 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
3168 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
Doptimized_ops.h832 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
2815 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
2946 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local