Home
last modified time | relevance | path

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

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DActivation.cpp47 constexpr uint32_t kOutputTensor = 0; variable
440 return context->setOutputShape(kOutputTensor, output); in prepare()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
456 context->getOutputShape(kOutputTensor)); in executeRelu()
461 context->getOutputShape(kOutputTensor)); in executeRelu()
480 context->getOutputShape(kOutputTensor)); in executeRelu1()
485 context->getOutputShape(kOutputTensor)); in executeRelu1()
509 context->getOutputShape(kOutputTensor)); in executeRelu6()
514 context->getOutputShape(kOutputTensor)); in executeRelu6()
577 context->getOutputShape(kOutputTensor)); in executeTanh()
[all …]
H A 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()
107 context->getOutputShape(kOutputTensor)); in execute()
113 context->getOutputShape(kOutputTensor)); in execute()
[all …]
H A 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()
134 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
142 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
150 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
[all …]
H A 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()
506 context->getOutputShape(kOutputTensor)); in executeAdd()
522 context->getOutputShape(kOutputTensor)); in executeAdd()
530 context->getOutputShape(kOutputTensor)); in executeAdd()
572 context->getOutputShape(kOutputTensor)); in executeMul()
580 context->getOutputShape(kOutputTensor)); in executeMul()
622 context->getOutputShape(kOutputTensor)); in executeSub()
[all …]
H A 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()
150 context->getOutputBuffer<float>(kOutputTensor), in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
164 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
165 context->getOutputShape(kOutputTensor)); in execute()
171 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
H A DSlice.cpp40 constexpr uint32_t kOutputTensor = 0; variable
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
139 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
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()
181 context->getOutputShape(kOutputTensor)); in execute()
189 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
H A 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 …]
H A 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()
H A 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()
94 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
103 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
112 context->getOutputShape(kOutputTensor)); in executeGreaterEqualTyped()
121 context->getOutputShape(kOutputTensor)); in executeGreaterTyped()
148 Shape output = context->getOutputShape(kOutputTensor); in prepare()
[all …]
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
327 context->getOutputShape(kOutputTensor)); in execute()
337 context->getOutputShape(kOutputTensor)); in execute()
346 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
H A 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)));
131 Shape outputShape(context->getOutputShape(kOutputTensor));
136 convertUInt8ToInt8(output_uint8, context->getOutputBuffer<int8_t>(kOutputTensor));
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()
H A DSoftmax.cpp48 constexpr uint32_t kOutputTensor = 0; variable
279 Shape output = context->getOutputShape(kOutputTensor); in prepare()
281 return context->setOutputShape(kOutputTensor, output); in prepare()
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()
313 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
H A 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 …]
H A 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()
H A DLogicalNot.cpp30 constexpr uint32_t kOutputTensor = 0; variable
57 Shape output = context->getOutputShape(kOutputTensor); in prepare()
59 return context->setOutputShape(kOutputTensor, output); in prepare()
65 context->getOutputBuffer<bool8>(kOutputTensor)); in execute()
H A 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 …]
H A 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()
H A DSelect.cpp33 constexpr uint32_t kOutputTensor = 0; variable
63 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
64 context->getOutputShape(kOutputTensor)); in executeTyped()
101 Shape output = context->getOutputShape(kOutputTensor); in prepare()
103 return context->setOutputShape(kOutputTensor, output); in prepare()
H A DSqueeze.cpp36 constexpr uint32_t kOutputTensor = 0; variable
73 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
124 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
135 context->getOutputBuffer(kOutputTensor), in execute()
136 context->getOutputShape(kOutputTensor)); in execute()
/aosp12/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
H A 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