Home
last modified time | relevance | path

Searched refs:kOutputTensor (Results 1 – 25 of 118) sorted by relevance

12345

/packages/modules/NeuralNetworks/common/cpu_operations/
DActivation.cpp374 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
394 return context->setOutputShape(kOutputTensor, output); in prepare()
399 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
404 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
405 context->getOutputShape(kOutputTensor)); in executeRelu()
409 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
410 context->getOutputShape(kOutputTensor)); in executeRelu()
414 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
415 context->getOutputShape(kOutputTensor)); in executeRelu()
419 context->getOutputBuffer<int8_t>(kOutputTensor), in executeRelu()
[all …]
DQuantize.cpp65 Shape output = context->getOutputShape(kOutputTensor); in prepare()
67 return context->setOutputShape(kOutputTensor, output); in prepare()
72 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
75 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
79 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
80 context->getOutputShape(kOutputTensor)); in execute()
83 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
84 context->getOutputShape(kOutputTensor)); in execute()
89 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
90 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DBroadcast.cpp443 Shape output = context->getOutputShape(kOutputTensor); in prepare()
447 return context->setOutputShape(kOutputTensor, output); in prepare()
452 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
460 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
461 context->getOutputShape(kOutputTensor)); in executeAdd()
468 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
469 context->getOutputShape(kOutputTensor)); in executeAdd()
476 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
477 context->getOutputShape(kOutputTensor)); in executeAdd()
484 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
[all …]
DDequantize.cpp77 Shape output = context->getOutputShape(kOutputTensor); in prepare()
79 return context->setOutputShape(kOutputTensor, output); in prepare()
84 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
87 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
94 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
96 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
102 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
104 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
110 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
112 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
[all …]
DElementwise.cpp78 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
82 context->getOutputBuffer<float>(kOutputTensor)); in execute()
96 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
100 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
105 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
118 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeRsqrt()
122 context->getOutputBuffer<float>(kOutputTensor)); in executeRsqrt()
125 const Shape outShape = context->getOutputShape(kOutputTensor); in executeRsqrt()
131 context->getOutputBuffer<uint8_t>(kOutputTensor)); in executeRsqrt()
135 const Shape outShape = context->getOutputShape(kOutputTensor); in executeRsqrt()
[all …]
DTranspose.cpp77 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
81 Shape output = context->getOutputShape(kOutputTensor); in prepare()
109 return context->setOutputShape(kOutputTensor, output); in prepare()
114 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
122 context->getOutputBuffer<float>(kOutputTensor), in execute()
123 context->getOutputShape(kOutputTensor)); in execute()
129 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
130 context->getOutputShape(kOutputTensor)); in execute()
136 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
137 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DSlice.cpp93 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
107 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
112 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
121 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
122 context->getOutputShape(kOutputTensor)); in execute()
130 context->getOutputBuffer<float>(kOutputTensor), in execute()
131 context->getOutputShape(kOutputTensor)); in execute()
139 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
140 context->getOutputShape(kOutputTensor)); in execute()
148 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DComparisons.cpp69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
70 context->getOutputShape(kOutputTensor)); in executeLessTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
79 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
88 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
97 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
106 context->getOutputShape(kOutputTensor)); in executeGreaterEqualTyped()
[all …]
DPRelu.cpp103 Shape output = context->getOutputShape(kOutputTensor); in prepare()
105 return context->setOutputShape(kOutputTensor, output); in prepare()
119 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
120 context->getOutputShape(kOutputTensor)); in execute()
130 context->getOutputBuffer<float>(kOutputTensor), in execute()
131 context->getOutputShape(kOutputTensor)); in execute()
137 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
138 context->getOutputShape(kOutputTensor)); in execute()
145 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
146 context->getOutputShape(kOutputTensor)); in execute()
DFullyConnected.cpp186 Shape output = context->getOutputShape(kOutputTensor); in prepare()
188 return context->setOutputShape(kOutputTensor, output); in prepare()
193 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
203 context->getOutputBuffer<float>(kOutputTensor), in execute()
204 context->getOutputShape(kOutputTensor)); in execute()
213 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
214 context->getOutputShape(kOutputTensor)); in execute()
223 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
224 context->getOutputShape(kOutputTensor)); in execute()
233 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DElu.cpp51 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
56 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
62 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
63 context->getOutputShape(kOutputTensor)); in execute()
68 context->getOutputBuffer<float>(kOutputTensor), in execute()
69 context->getOutputShape(kOutputTensor)); in execute()
DSoftmax.cpp238 Shape output = context->getOutputShape(kOutputTensor); in prepare()
240 return context->setOutputShape(kOutputTensor, output); in prepare()
245 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
254 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
255 context->getOutputShape(kOutputTensor)); in execute()
260 context->getOutputBuffer<float>(kOutputTensor), in execute()
261 context->getOutputShape(kOutputTensor)); in execute()
266 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
267 context->getOutputShape(kOutputTensor)); in execute()
272 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DLogicalAndOr.cpp60 Shape output = context->getOutputShape(kOutputTensor); in prepare()
62 return context->setOutputShape(kOutputTensor, output); in prepare()
69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
70 context->getOutputShape(kOutputTensor)); in executeAnd()
77 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
78 context->getOutputShape(kOutputTensor)); in executeOr()
DBatchMatmul.cpp223 Shape outputTensorShape = context->getOutputShape(kOutputTensor); in prepare()
226 return context->setOutputShape(kOutputTensor, outputTensorShape); in prepare()
238 context->getOutputBuffer<float>(kOutputTensor), in execute()
239 context->getOutputShape(kOutputTensor)); in execute()
247 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
248 context->getOutputShape(kOutputTensor)); in execute()
256 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
257 context->getOutputShape(kOutputTensor)); in execute()
265 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
266 context->getOutputShape(kOutputTensor)); in execute()
DGather.cpp58 Shape output = context->getOutputShape(kOutputTensor); in prepare()
69 return context->setOutputShape(kOutputTensor, output); in prepare()
81 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
87 context->getOutputBuffer<float>(kOutputTensor)); in execute()
93 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
99 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
105 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DNeg.cpp45 Shape output = context->getOutputShape(kOutputTensor); in prepare()
47 return context->setOutputShape(kOutputTensor, output); in prepare()
55 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
59 context->getOutputBuffer<float>(kOutputTensor)); in execute()
63 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
DResizeImageOps.cpp217 return context->setOutputShape(kOutputTensor, output); in prepare()
222 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
233 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
234 context->getOutputShape(kOutputTensor)); in execute()
238 halfPixelCenters, context->getOutputBuffer<float>(kOutputTensor), in execute()
239 context->getOutputShape(kOutputTensor)); in execute()
243 halfPixelCenters, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
244 context->getOutputShape(kOutputTensor)); in execute()
248 halfPixelCenters, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
249 context->getOutputShape(kOutputTensor)); in execute()
DConcatenation.cpp109 context->getOutputBuffer<T>(kOutputTensor), in concatenation()
110 context->getOutputShape(kOutputTensor)); in concatenation()
132 std::vector<uint8_t> output_uint8(getNumberOfElements(context->getOutputShape(kOutputTensor))); in concatenation()
133 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation()
138 convertUInt8ToInt8(output_uint8, context->getOutputBuffer<int8_t>(kOutputTensor)); in concatenation()
169 Shape output = context->getOutputShape(kOutputTensor); in prepare()
173 return context->setOutputShape(kOutputTensor, output); in prepare()
178 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DL2Normalization.cpp210 Shape output = context->getOutputShape(kOutputTensor); in prepare()
223 return context->setOutputShape(kOutputTensor, output); in prepare()
235 context->getOutputBuffer<float>(kOutputTensor), in execute()
236 context->getOutputShape(kOutputTensor)); in execute()
240 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
241 context->getOutputShape(kOutputTensor)); in execute()
245 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
246 context->getOutputShape(kOutputTensor)); in execute()
250 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
251 context->getOutputShape(kOutputTensor)); in execute()
DChannelShuffle.cpp59 return context->setOutputShape(kOutputTensor, input); in prepare()
70 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
74 context->getOutputBuffer<float>(kOutputTensor)); in execute()
78 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
82 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DTransposeConv2D.cpp484 Shape output = context->getOutputShape(kOutputTensor); in prepare()
491 return context->setOutputShape(kOutputTensor, output); in prepare()
496 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
507 context->getOutputBuffer<float>(kOutputTensor), in execute()
508 context->getOutputShape(kOutputTensor)); in execute()
516 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
517 context->getOutputShape(kOutputTensor)); in execute()
531 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
532 context->getOutputShape(kOutputTensor)); in execute()
540 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DFill.cpp31 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
32 const int numElements = getNumberOfElements(context->getOutputShape(kOutputTensor)); in executeTyped()
46 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
52 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
DRoiPooling.cpp222 return context->setOutputShape(kOutputTensor, output); in prepare()
237 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
238 context->getOutputShape(kOutputTensor)); in execute()
249 context->getOutputBuffer<float>(kOutputTensor), in execute()
250 context->getOutputShape(kOutputTensor)); in execute()
261 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
262 context->getOutputShape(kOutputTensor)); in execute()
273 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
274 context->getOutputShape(kOutputTensor)); in execute()
DDepthwiseConv2D.cpp465 Shape output = context->getOutputShape(kOutputTensor); in prepare()
472 return context->setOutputShape(kOutputTensor, output); in prepare()
477 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
492 context->getOutputBuffer<float>(kOutputTensor), in execute()
493 context->getOutputShape(kOutputTensor)); in execute()
505 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
506 context->getOutputShape(kOutputTensor)); in execute()
523 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
524 context->getOutputShape(kOutputTensor)); in execute()
536 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp48 constexpr uint32_t kOutputTensor = 0; variable
88 Shape output = context->getOutputShape(kOutputTensor); in prepare()
90 return context->setOutputShape(kOutputTensor, output); in prepare()
123 if (context->getOutputType(kOutputTensor) == OperandType::TENSOR_FLOAT32) { in execute()
124 float* output = context->getOutputBuffer<float>(kOutputTensor); in execute()
127 uint64_t* output = context->getOutputBuffer<uint64_t>(kOutputTensor); in execute()
128 Shape outputShape = context->getOutputShape(kOutputTensor); in execute()

12345