Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlstm_cell.h67 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
278 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
H A Dfully_connected.h188 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h1168 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCell() local
1355 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCellWithSymmetricRange() local
2238 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
3018 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
3170 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
H A Doptimized_ops.h770 const int weights_dim_count = weights_shape.DimensionsCount(); in ShuffledFullyConnected() local
2490 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local
2621 const int weights_dim_count = weights_shape.DimensionsCount(); in LstmCell() local