Home
last modified time | relevance | path

Searched defs:output_dim_count (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dfully_connected.h88 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnectedPerChannel() local
131 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
173 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dfully_connected.h87 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
134 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
187 const int output_dim_count = output_shape.DimensionsCount(); in ShuffledFullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dfully_connected.h51 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnectedPerChannel() local
123 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h784 const int output_dim_count = output_shape.DimensionsCount(); in LegacyFullyConnectedAsGEMVWorkerImpl() local
1053 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnectedAsGEMV() local
1112 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
1167 const int output_dim_count = output_shape.DimensionsCount(); in GEMVForLstmCell() local
1354 const int output_dim_count = output_shape.DimensionsCount(); in GEMVForLstmCellWithSymmetricRange() local
1662 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
1806 const int output_dim_count = output_shape.DimensionsCount(); in LegacyInt8FullyConnectedAsGEMVWorkerImpl() local
2064 const int output_dim_count = output_shape.DimensionsCount(); in LegacyInt8FullyConnectedAsGEMV() local
2128 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
2237 const int output_dim_count = output_shape.DimensionsCount(); in ShuffledFullyConnected() local
H A Doptimized_ops.h325 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
391 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
769 const int output_dim_count = output_shape.DimensionsCount(); in ShuffledFullyConnected() local