Home
last modified time | relevance | path

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

123

/frameworks/ml/nn/common/operations/
DActivation.cpp47 constexpr uint32_t kOutputTensor = 0; variable
410 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
430 return context->setOutputShape(kOutputTensor, output); in prepare()
435 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
440 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
441 context->getOutputShape(kOutputTensor)); in executeRelu()
445 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
446 context->getOutputShape(kOutputTensor)); in executeRelu()
450 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
[all …]
DQuantize.cpp36 constexpr uint32_t kOutputTensor = 0; variable
74 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
91 Shape output = context->getOutputShape(kOutputTensor); in prepare()
93 return context->setOutputShape(kOutputTensor, output); in prepare()
98 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
101 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
105 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
106 context->getOutputShape(kOutputTensor)); in execute()
109 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
110 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DDequantize.cpp32 constexpr uint32_t kOutputTensor = 0; variable
84 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
110 Shape output = context->getOutputShape(kOutputTensor); in prepare()
112 return context->setOutputShape(kOutputTensor, output); in prepare()
117 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
120 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
127 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
129 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
135 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
137 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
[all …]
DTranspose.cpp41 constexpr uint32_t kOutputTensor = 0; variable
101 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
105 Shape output = context->getOutputShape(kOutputTensor); in prepare()
133 return context->setOutputShape(kOutputTensor, output); in prepare()
138 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
146 context->getOutputBuffer<float>(kOutputTensor), in execute()
147 context->getOutputShape(kOutputTensor)); in execute()
153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
160 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DSlice.cpp38 constexpr uint32_t kOutputTensor = 0; variable
121 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
134 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
139 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
148 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
149 context->getOutputShape(kOutputTensor)); in execute()
157 context->getOutputBuffer<float>(kOutputTensor), in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
166 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
167 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DBroadcast.cpp50 constexpr uint32_t kOutputTensor = 0; variable
455 Shape output = context->getOutputShape(kOutputTensor); in validate()
482 Shape output = context->getOutputShape(kOutputTensor); in prepare()
486 return context->setOutputShape(kOutputTensor, output); in prepare()
491 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
499 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
500 context->getOutputShape(kOutputTensor)); in executeAdd()
507 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
508 context->getOutputShape(kOutputTensor)); in executeAdd()
515 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
[all …]
DElu.cpp40 constexpr uint32_t kOutputTensor = 0; variable
75 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
80 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
86 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
87 context->getOutputShape(kOutputTensor)); in execute()
92 context->getOutputBuffer<float>(kOutputTensor), in execute()
93 context->getOutputShape(kOutputTensor)); in execute()
DPRelu.cpp43 constexpr uint32_t kOutputTensor = 0; variable
123 Shape output = context->getOutputShape(kOutputTensor); in prepare()
125 return context->setOutputShape(kOutputTensor, output); in prepare()
139 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
140 context->getOutputShape(kOutputTensor)); in execute()
150 context->getOutputBuffer<float>(kOutputTensor), in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
157 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
165 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DNeg.cpp36 constexpr uint32_t kOutputTensor = 0; variable
67 Shape output = context->getOutputShape(kOutputTensor); in prepare()
69 return context->setOutputShape(kOutputTensor, output); in prepare()
77 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
81 context->getOutputBuffer<float>(kOutputTensor)); in execute()
85 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
DComparisons.cpp36 constexpr uint32_t kOutputTensor = 0; variable
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
79 context->getOutputShape(kOutputTensor)); in executeLessTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
88 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
97 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
106 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
114 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DLogicalAndOr.cpp33 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()
DElementwise.cpp34 constexpr uint32_t kOutputTensor = 0; variable
55 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
59 context->getOutputBuffer<float>(kOutputTensor)); in execute()
73 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
77 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
82 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
136 Shape output = context->getOutputShape(kOutputTensor); in prepare()
138 return context->setOutputShape(kOutputTensor, output); in prepare()
143 Shape output = context->getOutputShape(kOutputTensor); in prepareFloor()
146 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
DFill.cpp32 constexpr uint32_t kOutputTensor = 0; variable
40 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
41 const int numElements = getNumberOfElements(context->getOutputShape(kOutputTensor)); in executeTyped()
72 OperandType outputType = context->getOutputType(kOutputTensor); in validate()
90 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
96 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
DGather.cpp36 constexpr uint32_t kOutputTensor = 0; variable
90 Shape output = context->getOutputShape(kOutputTensor); in prepare()
101 return context->setOutputShape(kOutputTensor, output); in prepare()
113 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
119 context->getOutputBuffer<float>(kOutputTensor)); in execute()
125 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
131 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
137 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DFullyConnected.cpp44 constexpr uint32_t kOutputTensor = 0; variable
251 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
296 Shape output = context->getOutputShape(kOutputTensor); in prepare()
298 return context->setOutputShape(kOutputTensor, output); in prepare()
303 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
313 context->getOutputBuffer<float>(kOutputTensor), in execute()
314 context->getOutputShape(kOutputTensor)); in execute()
323 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
324 context->getOutputShape(kOutputTensor)); in execute()
333 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DConcatenation.cpp41 constexpr uint32_t kOutputTensor = 0; variable
105 context->getOutputBuffer<T>(kOutputTensor), in concatenation()
106 context->getOutputShape(kOutputTensor)); in concatenation()
128 std::vector<uint8_t> output_uint8(getNumberOfElements(context->getOutputShape(kOutputTensor))); in concatenation()
129 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation()
134 convertUInt8ToInt8(output_uint8, context->getOutputBuffer<int8_t>(kOutputTensor)); in concatenation()
159 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
200 Shape output = context->getOutputShape(kOutputTensor); in prepare()
204 return context->setOutputShape(kOutputTensor, output); in prepare()
209 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DLogicalNot.cpp31 constexpr uint32_t kOutputTensor = 0; variable
60 Shape output = context->getOutputShape(kOutputTensor); in prepare()
62 return context->setOutputShape(kOutputTensor, output); in prepare()
68 context->getOutputBuffer<bool8>(kOutputTensor)); in execute()
DSoftmax.cpp44 constexpr uint32_t kOutputTensor = 0; variable
272 Shape output = context->getOutputShape(kOutputTensor); in prepare()
274 return context->setOutputShape(kOutputTensor, output); in prepare()
279 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
288 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
289 context->getOutputShape(kOutputTensor)); in execute()
294 context->getOutputBuffer<float>(kOutputTensor), in execute()
295 context->getOutputShape(kOutputTensor)); in execute()
300 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
301 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DChannelShuffle.cpp38 constexpr uint32_t kOutputTensor = 0; variable
92 return context->setOutputShape(kOutputTensor, input); in prepare()
103 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
107 context->getOutputBuffer<float>(kOutputTensor)); in execute()
111 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
115 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
DL2Normalization.cpp41 constexpr uint32_t kOutputTensor = 0; variable
241 Shape output = context->getOutputShape(kOutputTensor); in prepare()
254 return context->setOutputShape(kOutputTensor, output); in prepare()
266 context->getOutputBuffer<float>(kOutputTensor), in execute()
267 context->getOutputShape(kOutputTensor)); in execute()
271 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
272 context->getOutputShape(kOutputTensor)); in execute()
276 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
277 context->getOutputShape(kOutputTensor)); in execute()
281 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DSqueeze.cpp37 constexpr uint32_t kOutputTensor = 0; variable
75 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
126 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
137 context->getOutputBuffer(kOutputTensor), in execute()
138 context->getOutputShape(kOutputTensor)); in execute()
DInstanceNormalization.cpp41 constexpr uint32_t kOutputTensor = 0; variable
128 return context->setOutputShape(kOutputTensor, input); in prepare()
140 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
141 context->getOutputShape(kOutputTensor)); in execute()
149 context->getOutputBuffer<float>(kOutputTensor), in execute()
150 context->getOutputShape(kOutputTensor)); in execute()
DResizeImageOps.cpp48 constexpr uint32_t kOutputTensor = 0; variable
272 return context->setOutputShape(kOutputTensor, output); in prepare()
277 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
288 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
289 context->getOutputShape(kOutputTensor)); in execute()
293 halfPixelCenters, context->getOutputBuffer<float>(kOutputTensor), in execute()
294 context->getOutputShape(kOutputTensor)); in execute()
298 halfPixelCenters, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
299 context->getOutputShape(kOutputTensor)); in execute()
303 halfPixelCenters, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
[all …]
DSelect.cpp34 constexpr uint32_t kOutputTensor = 0; variable
66 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
67 context->getOutputShape(kOutputTensor)); in executeTyped()
104 Shape output = context->getOutputShape(kOutputTensor); in prepare()
106 return context->setOutputShape(kOutputTensor, output); in prepare()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp49 constexpr uint32_t kOutputTensor = 0; variable
89 Shape output = context->getOutputShape(kOutputTensor); in prepare()
91 return context->setOutputShape(kOutputTensor, output); in prepare()
124 if (context->getOutputType(kOutputTensor) == OperandType::TENSOR_FLOAT32) { in execute()
125 float* output = context->getOutputBuffer<float>(kOutputTensor); in execute()
128 uint64_t* output = context->getOutputBuffer<uint64_t>(kOutputTensor); in execute()
129 Shape outputShape = context->getOutputShape(kOutputTensor); in execute()

123