Home
last modified time | relevance | path

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

123

/packages/modules/NeuralNetworks/common/operations/
DActivation.cpp47 constexpr uint32_t kOutputTensor = 0; variable
420 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
440 return context->setOutputShape(kOutputTensor, output); in prepare()
445 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
450 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
455 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
456 context->getOutputShape(kOutputTensor)); in executeRelu()
460 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
461 context->getOutputShape(kOutputTensor)); in executeRelu()
[all …]
DBroadcast.cpp52 constexpr uint32_t kOutputTensor = 0; variable
459 Shape output = context->getOutputShape(kOutputTensor); in validate()
488 Shape output = context->getOutputShape(kOutputTensor); in prepare()
492 return context->setOutputShape(kOutputTensor, output); in prepare()
497 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
505 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
506 context->getOutputShape(kOutputTensor)); in executeAdd()
513 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
514 context->getOutputShape(kOutputTensor)); in executeAdd()
521 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
[all …]
DQuantize.cpp35 constexpr uint32_t kOutputTensor = 0; variable
71 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
88 Shape output = context->getOutputShape(kOutputTensor); in prepare()
90 return context->setOutputShape(kOutputTensor, output); in prepare()
95 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
98 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
102 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
103 context->getOutputShape(kOutputTensor)); in execute()
106 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
107 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DDequantize.cpp31 constexpr uint32_t kOutputTensor = 0; variable
83 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
109 Shape output = context->getOutputShape(kOutputTensor); in prepare()
111 return context->setOutputShape(kOutputTensor, output); in prepare()
116 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
119 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
126 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
128 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
134 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
136 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
[all …]
DTranspose.cpp42 constexpr uint32_t kOutputTensor = 0; variable
105 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
109 Shape output = context->getOutputShape(kOutputTensor); in prepare()
137 return context->setOutputShape(kOutputTensor, output); in prepare()
142 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
150 context->getOutputBuffer<float>(kOutputTensor), in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
157 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
164 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DSlice.cpp40 constexpr uint32_t kOutputTensor = 0; variable
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
139 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
144 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
162 context->getOutputBuffer<float>(kOutputTensor), in execute()
163 context->getOutputShape(kOutputTensor)); in execute()
171 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
172 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DComparisons.cpp35 constexpr uint32_t kOutputTensor = 0; variable
75 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
76 context->getOutputShape(kOutputTensor)); in executeLessTyped()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
85 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
93 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
94 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
102 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
103 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
111 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DPRelu.cpp42 constexpr uint32_t kOutputTensor = 0; variable
125 Shape output = context->getOutputShape(kOutputTensor); in prepare()
127 return context->setOutputShape(kOutputTensor, output); in prepare()
141 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
142 context->getOutputShape(kOutputTensor)); in execute()
152 context->getOutputBuffer<float>(kOutputTensor), in execute()
153 context->getOutputShape(kOutputTensor)); in execute()
159 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
160 context->getOutputShape(kOutputTensor)); in execute()
167 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DElementwise.cpp33 constexpr uint32_t kOutputTensor = 0; variable
52 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
56 context->getOutputBuffer<float>(kOutputTensor)); in execute()
70 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
74 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
79 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
130 Shape output = context->getOutputShape(kOutputTensor); in prepare()
132 return context->setOutputShape(kOutputTensor, output); in prepare()
137 Shape output = context->getOutputShape(kOutputTensor); in prepareFloor()
140 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
DFullyConnected.cpp46 constexpr uint32_t kOutputTensor = 0; variable
254 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
299 Shape output = context->getOutputShape(kOutputTensor); in prepare()
301 return context->setOutputShape(kOutputTensor, output); in prepare()
306 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
316 context->getOutputBuffer<float>(kOutputTensor), in execute()
317 context->getOutputShape(kOutputTensor)); in execute()
326 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
327 context->getOutputShape(kOutputTensor)); in execute()
336 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DElu.cpp37 constexpr uint32_t kOutputTensor = 0; variable
74 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
79 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
85 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
86 context->getOutputShape(kOutputTensor)); in execute()
91 context->getOutputBuffer<float>(kOutputTensor), in execute()
92 context->getOutputShape(kOutputTensor)); in execute()
DSoftmax.cpp48 constexpr uint32_t kOutputTensor = 0; variable
279 Shape output = context->getOutputShape(kOutputTensor); in prepare()
281 return context->setOutputShape(kOutputTensor, output); in prepare()
286 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
295 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
296 context->getOutputShape(kOutputTensor)); in execute()
301 context->getOutputBuffer<float>(kOutputTensor), in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
307 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
308 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DConcatenation.cpp44 constexpr uint32_t kOutputTensor = 0; variable
107 context->getOutputBuffer<T>(kOutputTensor), in concatenation()
108 context->getOutputShape(kOutputTensor)); in concatenation()
130 std::vector<uint8_t> output_uint8(getNumberOfElements(context->getOutputShape(kOutputTensor))); in concatenation()
131 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation()
136 convertUInt8ToInt8(output_uint8, context->getOutputBuffer<int8_t>(kOutputTensor)); in concatenation()
162 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
206 Shape output = context->getOutputShape(kOutputTensor); in prepare()
210 return context->setOutputShape(kOutputTensor, output); in prepare()
215 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DLogicalAndOr.cpp35 constexpr uint32_t kOutputTensor = 0; variable
77 Shape output = context->getOutputShape(kOutputTensor); in prepare()
79 return context->setOutputShape(kOutputTensor, output); in prepare()
86 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
87 context->getOutputShape(kOutputTensor)); in executeAnd()
94 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
95 context->getOutputShape(kOutputTensor)); in executeOr()
DGather.cpp35 constexpr uint32_t kOutputTensor = 0; variable
87 Shape output = context->getOutputShape(kOutputTensor); in prepare()
98 return context->setOutputShape(kOutputTensor, output); in prepare()
110 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
116 context->getOutputBuffer<float>(kOutputTensor)); in execute()
122 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
128 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
134 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DL2Normalization.cpp43 constexpr uint32_t kOutputTensor = 0; variable
246 Shape output = context->getOutputShape(kOutputTensor); in prepare()
259 return context->setOutputShape(kOutputTensor, output); in prepare()
271 context->getOutputBuffer<float>(kOutputTensor), in execute()
272 context->getOutputShape(kOutputTensor)); in execute()
276 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
277 context->getOutputShape(kOutputTensor)); in execute()
281 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
282 context->getOutputShape(kOutputTensor)); in execute()
286 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DNeg.cpp35 constexpr uint32_t kOutputTensor = 0; variable
64 Shape output = context->getOutputShape(kOutputTensor); in prepare()
66 return context->setOutputShape(kOutputTensor, output); in prepare()
74 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
78 context->getOutputBuffer<float>(kOutputTensor)); in execute()
82 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
DResizeImageOps.cpp49 constexpr uint32_t kOutputTensor = 0; variable
278 return context->setOutputShape(kOutputTensor, output); in prepare()
283 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
294 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
295 context->getOutputShape(kOutputTensor)); in execute()
299 halfPixelCenters, context->getOutputBuffer<float>(kOutputTensor), in execute()
300 context->getOutputShape(kOutputTensor)); in execute()
304 halfPixelCenters, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
305 context->getOutputShape(kOutputTensor)); in execute()
309 halfPixelCenters, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DFill.cpp31 constexpr uint32_t kOutputTensor = 0; variable
37 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
38 const int numElements = getNumberOfElements(context->getOutputShape(kOutputTensor)); in executeTyped()
69 OperandType outputType = context->getOutputType(kOutputTensor); in validate()
87 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
93 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
DTransposeConv2D.cpp47 constexpr uint32_t kOutputTensor = 0; variable
536 Shape output = context->getOutputShape(kOutputTensor); in prepare()
543 return context->setOutputShape(kOutputTensor, output); in prepare()
548 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
559 context->getOutputBuffer<float>(kOutputTensor), in execute()
560 context->getOutputShape(kOutputTensor)); in execute()
568 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
569 context->getOutputShape(kOutputTensor)); in execute()
583 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
584 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DChannelShuffle.cpp35 constexpr uint32_t kOutputTensor = 0; variable
89 return context->setOutputShape(kOutputTensor, input); in prepare()
100 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
104 context->getOutputBuffer<float>(kOutputTensor)); in execute()
108 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
112 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DDepthwiseConv2D.cpp44 constexpr uint32_t kOutputTensor = 0; variable
474 const float outputScale = context->getInputShape(kOutputTensor).scale; in validate()
568 Shape output = context->getOutputShape(kOutputTensor); in prepare()
575 return context->setOutputShape(kOutputTensor, output); in prepare()
580 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
595 context->getOutputBuffer<float>(kOutputTensor), in execute()
596 context->getOutputShape(kOutputTensor)); in execute()
608 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
609 context->getOutputShape(kOutputTensor)); in execute()
626 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DRoiPooling.cpp49 constexpr uint32_t kOutputTensor = 0; variable
274 return context->setOutputShape(kOutputTensor, output); in prepare()
289 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
290 context->getOutputShape(kOutputTensor)); in execute()
301 context->getOutputBuffer<float>(kOutputTensor), in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
313 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
314 context->getOutputShape(kOutputTensor)); in execute()
325 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
326 context->getOutputShape(kOutputTensor)); in execute()
DInstanceNormalization.cpp43 constexpr uint32_t kOutputTensor = 0; variable
130 return context->setOutputShape(kOutputTensor, input); in prepare()
142 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
143 context->getOutputShape(kOutputTensor)); in execute()
151 context->getOutputBuffer<float>(kOutputTensor), in execute()
152 context->getOutputShape(kOutputTensor)); in execute()
/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()

123