Home
last modified time | relevance | path

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

123

/frameworks/ml/nn/common/operations/
DActivation.cpp35 constexpr uint32_t kOutputTensor = 0; variable
251 return context->setOutputShape(kOutputTensor, output); in prepare()
256 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
261 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
262 context->getOutputShape(kOutputTensor)); in executeRelu()
266 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
267 context->getOutputShape(kOutputTensor)); in executeRelu()
271 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
272 context->getOutputShape(kOutputTensor)); in executeRelu()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
[all …]
DQuantize.cpp35 constexpr uint32_t kOutputTensor = 0; variable
70 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
82 Shape output = context->getOutputShape(kOutputTensor); in prepare()
84 return context->setOutputShape(kOutputTensor, output); in prepare()
89 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
94 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
95 context->getOutputShape(kOutputTensor)); in execute()
98 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
99 context->getOutputShape(kOutputTensor)); in execute()
103 << " output type: " << toString(context->getOutputType(kOutputTensor)) in execute()
DDequantize.cpp32 constexpr uint32_t kOutputTensor = 0; variable
82 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
101 Shape output = context->getOutputShape(kOutputTensor); in prepare()
103 return context->setOutputShape(kOutputTensor, output); in prepare()
108 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
111 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
118 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
120 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
126 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
128 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
[all …]
DTranspose.cpp36 constexpr uint32_t kOutputTensor = 0; variable
88 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
92 Shape output = context->getOutputShape(kOutputTensor); in prepare()
120 return context->setOutputShape(kOutputTensor, output); in prepare()
125 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
133 context->getOutputBuffer<float>(kOutputTensor), in execute()
134 context->getOutputShape(kOutputTensor)); in execute()
140 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
141 context->getOutputShape(kOutputTensor)); in execute()
147 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DBroadcast.cpp41 constexpr uint32_t kOutputTensor = 0; variable
367 Shape output = context->getOutputShape(kOutputTensor); in validate()
385 Shape output = context->getOutputShape(kOutputTensor); in prepare()
389 return context->setOutputShape(kOutputTensor, output); in prepare()
394 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
402 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
403 context->getOutputShape(kOutputTensor)); in executeAdd()
410 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
411 context->getOutputShape(kOutputTensor)); in executeAdd()
418 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
[all …]
DSlice.cpp34 constexpr uint32_t kOutputTensor = 0; variable
109 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
122 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
127 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
136 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
137 context->getOutputShape(kOutputTensor)); in execute()
145 context->getOutputBuffer<float>(kOutputTensor), in execute()
146 context->getOutputShape(kOutputTensor)); in execute()
154 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
155 context->getOutputShape(kOutputTensor)); in execute()
[all …]
DComparisons.cpp33 constexpr uint32_t kOutputTensor = 0; variable
72 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
73 context->getOutputShape(kOutputTensor)); in executeLessTyped()
81 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
82 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
90 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
91 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
99 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
100 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
108 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DNeg.cpp36 constexpr uint32_t kOutputTensor = 0; variable
65 Shape output = context->getOutputShape(kOutputTensor); in prepare()
67 return context->setOutputShape(kOutputTensor, output); in prepare()
75 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
79 context->getOutputBuffer<float>(kOutputTensor)); in execute()
83 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
DLogicalAndOr.cpp33 constexpr uint32_t kOutputTensor = 0; variable
75 Shape output = context->getOutputShape(kOutputTensor); in prepare()
77 return context->setOutputShape(kOutputTensor, output); in prepare()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
85 context->getOutputShape(kOutputTensor)); in executeAnd()
92 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
93 context->getOutputShape(kOutputTensor)); in executeOr()
DPRelu.cpp37 constexpr uint32_t kOutputTensor = 0; variable
112 Shape output = context->getOutputShape(kOutputTensor); in prepare()
114 return context->setOutputShape(kOutputTensor, output); in prepare()
128 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
129 context->getOutputShape(kOutputTensor)); in execute()
139 context->getOutputBuffer<float>(kOutputTensor), in execute()
140 context->getOutputShape(kOutputTensor)); in execute()
146 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
147 context->getOutputShape(kOutputTensor)); in execute()
DFullyConnected.cpp38 constexpr uint32_t kOutputTensor = 0; variable
169 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
220 Shape output = context->getOutputShape(kOutputTensor); in prepare()
223 return context->setOutputShape(kOutputTensor, output); in prepare()
228 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
238 context->getOutputBuffer<float>(kOutputTensor), in execute()
239 context->getOutputShape(kOutputTensor)); in execute()
248 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
249 context->getOutputShape(kOutputTensor)); in execute()
258 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
[all …]
DGather.cpp36 constexpr uint32_t kOutputTensor = 0; variable
82 Shape output = context->getOutputShape(kOutputTensor); in prepare()
93 return context->setOutputShape(kOutputTensor, output); in prepare()
105 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
111 context->getOutputBuffer<float>(kOutputTensor)); in execute()
117 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
123 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
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()
DResizeImageOps.cpp42 constexpr uint32_t kOutputTensor = 0; variable
189 return context->setOutputShape(kOutputTensor, output); in prepare()
194 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
203 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
204 context->getOutputShape(kOutputTensor)); in execute()
208 context->getOutputBuffer<float>(kOutputTensor), in execute()
209 context->getOutputShape(kOutputTensor)); in execute()
213 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
214 context->getOutputShape(kOutputTensor)); in execute()
DL2Normalization.cpp35 constexpr uint32_t kOutputTensor = 0; variable
177 Shape output = context->getOutputShape(kOutputTensor); in prepare()
187 return context->setOutputShape(kOutputTensor, output); in prepare()
199 context->getOutputBuffer<float>(kOutputTensor), in execute()
200 context->getOutputShape(kOutputTensor)); in execute()
204 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
205 context->getOutputShape(kOutputTensor)); in execute()
209 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
210 context->getOutputShape(kOutputTensor)); in execute()
DElementwise.cpp34 constexpr uint32_t kOutputTensor = 0; variable
52 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
56 context->getOutputBuffer<float>(kOutputTensor)); in execute()
78 Shape output = context->getOutputShape(kOutputTensor); in prepare()
80 return context->setOutputShape(kOutputTensor, output); in prepare()
DSoftmax.cpp38 constexpr uint32_t kOutputTensor = 0; variable
261 Shape output = context->getOutputShape(kOutputTensor); in prepare()
263 return context->setOutputShape(kOutputTensor, output); in prepare()
268 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
277 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
278 context->getOutputShape(kOutputTensor)); in execute()
283 context->getOutputBuffer<float>(kOutputTensor), in execute()
284 context->getOutputShape(kOutputTensor)); in execute()
289 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
290 context->getOutputShape(kOutputTensor)); in execute()
DConcatenation.cpp32 constexpr uint32_t kOutputTensor = 0; variable
94 context->getOutputBuffer<T>(kOutputTensor), in concatenation()
95 context->getOutputShape(kOutputTensor)); in concatenation()
116 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
150 Shape output = context->getOutputShape(kOutputTensor); in prepare()
154 return context->setOutputShape(kOutputTensor, output); in prepare()
159 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DSelect.cpp34 constexpr uint32_t kOutputTensor = 0; variable
63 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
64 context->getOutputShape(kOutputTensor)); in executeTyped()
99 Shape output = context->getOutputShape(kOutputTensor); in prepare()
101 return context->setOutputShape(kOutputTensor, output); in prepare()
DChannelShuffle.cpp36 constexpr uint32_t kOutputTensor = 0; variable
81 return context->setOutputShape(kOutputTensor, input); in prepare()
92 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
96 context->getOutputBuffer<float>(kOutputTensor)); in execute()
100 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
DInstanceNormalization.cpp41 constexpr uint32_t kOutputTensor = 0; variable
115 return context->setOutputShape(kOutputTensor, input); in prepare()
127 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
128 context->getOutputShape(kOutputTensor)); in execute()
136 context->getOutputBuffer<float>(kOutputTensor), in execute()
137 context->getOutputShape(kOutputTensor)); in execute()
DUnidirectionalSequenceRNN.cpp36 constexpr uint32_t kOutputTensor = 0; variable
68 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
69 Shape outputShape = context->getOutputShape(kOutputTensor); in executeTyped()
109 context->getOutputBuffer<T>(kOutputTensor)); in executeTyped()
160 Shape output = context->getOutputShape(kOutputTensor); in prepare()
165 return context->setOutputShape(kOutputTensor, output); in prepare()
DLogSoftmax.cpp38 constexpr uint32_t kOutputTensor = 0; variable
94 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
105 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
110 context->getOutputBuffer<float>(kOutputTensor)); in execute()
DRoiPooling.cpp43 constexpr uint32_t kOutputTensor = 0; variable
246 return context->setOutputShape(kOutputTensor, output); in prepare()
261 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
262 context->getOutputShape(kOutputTensor)); in execute()
273 context->getOutputBuffer<float>(kOutputTensor), in execute()
274 context->getOutputShape(kOutputTensor)); in execute()
285 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
286 context->getOutputShape(kOutputTensor)); in execute()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp46 constexpr uint32_t kOutputTensor = 0; variable
86 Shape output = context->getOutputShape(kOutputTensor); in prepare()
88 return context->setOutputShape(kOutputTensor, output); in prepare()
121 if (context->getOutputType(kOutputTensor) == OperandType::TENSOR_FLOAT32) { in execute()
122 float* output = context->getOutputBuffer<float>(kOutputTensor); in execute()
125 uint64_t* output = context->getOutputBuffer<uint64_t>(kOutputTensor); in execute()
126 Shape outputShape = context->getOutputShape(kOutputTensor); in execute()

123