Home
last modified time | relevance | path

Searched refs:tranche_depth (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h69 const int tranche_depth = in MaxPool() local
84 tranche_depth * sizeof(acc[0])); in MaxPool()
96 for (; channel <= tranche_depth - 16; channel += 16) { in MaxPool()
104 for (; channel <= tranche_depth - 8; channel += 8) { in MaxPool()
112 for (; channel < tranche_depth; ++channel) { in MaxPool()
122 for (; channel <= tranche_depth - 16; channel += 16) { in MaxPool()
128 for (; channel <= tranche_depth - 8; channel += 8) { in MaxPool()
135 for (; channel < tranche_depth; ++channel) { in MaxPool()
179 const int tranche_depth = in AveragePool() local
195 memset(acc, 0, tranche_depth * sizeof(acc[0])); in AveragePool()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3268 const int tranche_depth = in AveragePool() local
3284 memset(acc, 0, tranche_depth * sizeof(acc[0])); in AveragePool()
3296 for (; channel <= tranche_depth - 16; channel += 16) { in AveragePool()
3310 for (; channel <= tranche_depth - 8; channel += 8) { in AveragePool()
3323 for (; channel < tranche_depth; ++channel) { in AveragePool()
3335 for (; channel <= tranche_depth - 8; channel += 8) { \ in AveragePool()
3349 for (; channel <= tranche_depth - 8; channel += 8) { in AveragePool()
3360 for (; channel < tranche_depth; ++channel) { in AveragePool()
3458 const int tranche_depth = in MaxPool() local
3472 memset(acc, 0, tranche_depth * sizeof(acc[0])); in MaxPool()
[all …]