Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h84 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
86 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
88 output_shape, output_dim_count - 1); in FullyConnected()
131 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
133 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
135 output_shape, output_dim_count - 1); in FullyConnected()
184 const int output_dim_count = output_shape.DimensionsCount(); in ShuffledFullyConnected() local
186 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
188 output_shape, output_dim_count - 1); in ShuffledFullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h52 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
54 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
58 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h783 const int output_dim_count = output_shape.DimensionsCount(); in LegacyFullyConnectedAsGEMVWorkerImpl() local
784 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in LegacyFullyConnectedAsGEMVWorkerImpl()
1052 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnectedAsGEMV() local
1053 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedAsGEMV()
1054 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnectedAsGEMV()
1111 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
1113 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
1117 output_shape, output_dim_count - 1); in FullyConnected()
1130 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected()
1166 const int output_dim_count = output_shape.DimensionsCount(); in GEMVForLstmCell() local
[all …]
Doptimized_ops.h389 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
391 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
395 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected()
455 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected() local
457 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
459 output_shape, output_dim_count - 1); in FullyConnected()
833 const int output_dim_count = output_shape.DimensionsCount(); in ShuffledFullyConnected() local
835 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
837 output_shape, output_dim_count - 1); in ShuffledFullyConnected()