Searched refs:extended_output_shape (Results 1 – 7 of 7) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
| A D | prelu.h | 35 const RuntimeShape extended_output_shape = in BroadcastPrelu4DSlow() local
41 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastPrelu4DSlow()
42 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastPrelu4DSlow()
43 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastPrelu4DSlow()
44 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastPrelu4DSlow()
45 int output_index = Offset(extended_output_shape, b, y, x, c); in BroadcastPrelu4DSlow()
|
| A D | add.h | 206 const RuntimeShape extended_output_shape = variable
223 for (int b = 0; b < extended_output_shape.Dims(0); ++b) {
224 for (int y = 0; y < extended_output_shape.Dims(1); ++y) {
225 for (int x = 0; x < extended_output_shape.Dims(2); ++x) {
226 for (int c = 0; c < extended_output_shape.Dims(3); ++c) {
227 output_data[Offset(extended_output_shape, b, y, x, c)] =
251 const RuntimeShape extended_output_shape = in BroadcastAdd4DSlow() local
265 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastAdd4DSlow()
266 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastAdd4DSlow()
267 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastAdd4DSlow()
[all …]
|
| A D | mul.h | 85 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow() local
88 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastMul4DSlow()
89 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastMul4DSlow()
90 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastMul4DSlow()
91 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastMul4DSlow()
106 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastMul4DSlow()
|
| A D | sub.h | 489 const RuntimeShape extended_output_shape = in Sub() local
503 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in Sub()
504 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in Sub()
505 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in Sub()
506 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in Sub()
507 output_data[Offset(extended_output_shape, b, y, x, c)] = in Sub()
|
| A D | reference_ops.h | 1234 const RuntimeShape extended_output_shape = in BroadcastSelect4DSlow() local
1255 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastSelect4DSlow()
1256 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastSelect4DSlow()
1257 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastSelect4DSlow()
1258 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastSelect4DSlow()
1263 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastSelect4DSlow()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| A D | mul.h | 100 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow() local
103 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastMul4DSlow()
104 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastMul4DSlow()
105 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastMul4DSlow()
106 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastMul4DSlow()
121 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastMul4DSlow()
|
| A D | add.h | 59 const RuntimeShape extended_output_shape = in BroadcastBinaryFunction4DSlow() local
73 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastBinaryFunction4DSlow()
74 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastBinaryFunction4DSlow()
75 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastBinaryFunction4DSlow()
76 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastBinaryFunction4DSlow()
77 output_data[Offset(extended_output_shape, b, y, x, c)] = binary_func( in BroadcastBinaryFunction4DSlow()
|
Completed in 26 milliseconds