Home
last modified time | relevance | path

Searched refs:getOutputShape (Results 1 – 25 of 34) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DActivation.cpp256 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
262 context->getOutputShape(kOutputTensor)); in executeRelu()
267 context->getOutputShape(kOutputTensor)); in executeRelu()
272 context->getOutputShape(kOutputTensor)); in executeRelu()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
286 context->getOutputShape(kOutputTensor)); in executeRelu1()
291 context->getOutputShape(kOutputTensor)); in executeRelu1()
296 context->getOutputShape(kOutputTensor)); in executeRelu1()
304 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu6()
310 context->getOutputShape(kOutputTensor)); in executeRelu6()
[all …]
DQuantize.cpp82 Shape output = context->getOutputShape(kOutputTensor); in prepare()
89 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
95 context->getOutputShape(kOutputTensor)); in execute()
99 context->getOutputShape(kOutputTensor)); in execute()
DBroadcast.cpp367 Shape output = context->getOutputShape(kOutputTensor); in validate()
385 Shape output = context->getOutputShape(kOutputTensor); in prepare()
394 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
403 context->getOutputShape(kOutputTensor)); in executeAdd()
411 context->getOutputShape(kOutputTensor)); in executeAdd()
419 context->getOutputShape(kOutputTensor)); in executeAdd()
427 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
436 context->getOutputShape(kOutputTensor)); in executeMul()
444 context->getOutputShape(kOutputTensor)); in executeMul()
452 context->getOutputShape(kOutputTensor)); in executeMul()
[all …]
DTranspose.cpp92 Shape output = context->getOutputShape(kOutputTensor); in prepare()
125 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
134 context->getOutputShape(kOutputTensor)); in execute()
141 context->getOutputShape(kOutputTensor)); in execute()
148 context->getOutputShape(kOutputTensor)); in execute()
DSlice.cpp109 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
127 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
137 context->getOutputShape(kOutputTensor)); in execute()
146 context->getOutputShape(kOutputTensor)); in execute()
155 context->getOutputShape(kOutputTensor)); in execute()
164 context->getOutputShape(kOutputTensor)); in execute()
DGenerateProposals.cpp198 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
236 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
248 context->getOutputShape(kOutputTensor)); in execute()
260 context->getOutputShape(kOutputTensor)); in execute()
272 context->getOutputShape(kOutputTensor)); in execute()
522 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
526 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
530 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
534 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
676 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
[all …]
DComparisons.cpp73 context->getOutputShape(kOutputTensor)); in executeLessTyped()
82 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
91 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
100 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
109 context->getOutputShape(kOutputTensor)); in executeGreaterEqualTyped()
118 context->getOutputShape(kOutputTensor)); in executeGreaterTyped()
140 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DHeatmapMaxKeypoint.cpp245 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()
250 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare()
268 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute()
270 const auto outputKeypointShape = context->getOutputShape(kOutputKeypointTensor); in execute()
291 context->getOutputShape(kOutputScoreTensor), in execute()
293 context->getOutputShape(kOutputKeypointTensor), 1e-5f, in execute()
303 context->getOutputShape(kOutputScoreTensor), in execute()
305 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
DFullyConnected.cpp169 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
220 Shape output = context->getOutputShape(kOutputTensor); in prepare()
228 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
239 context->getOutputShape(kOutputTensor)); in execute()
249 context->getOutputShape(kOutputTensor)); in execute()
259 context->getOutputShape(kOutputTensor)); in execute()
DLogicalAndOr.cpp75 Shape output = context->getOutputShape(kOutputTensor); in prepare()
85 context->getOutputShape(kOutputTensor)); in executeAnd()
93 context->getOutputShape(kOutputTensor)); in executeOr()
DPRelu.cpp112 Shape output = context->getOutputShape(kOutputTensor); in prepare()
129 context->getOutputShape(kOutputTensor)); in execute()
140 context->getOutputShape(kOutputTensor)); in execute()
147 context->getOutputShape(kOutputTensor)); in execute()
DConcatenation.cpp95 context->getOutputShape(kOutputTensor)); in concatenation()
116 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
150 Shape output = context->getOutputShape(kOutputTensor); in prepare()
159 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DSoftmax.cpp261 Shape output = context->getOutputShape(kOutputTensor); in prepare()
268 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
278 context->getOutputShape(kOutputTensor)); in execute()
284 context->getOutputShape(kOutputTensor)); in execute()
290 context->getOutputShape(kOutputTensor)); in execute()
DL2Normalization.cpp177 Shape output = context->getOutputShape(kOutputTensor); in prepare()
200 context->getOutputShape(kOutputTensor)); in execute()
205 context->getOutputShape(kOutputTensor)); in execute()
210 context->getOutputShape(kOutputTensor)); in execute()
DResizeImageOps.cpp194 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
204 context->getOutputShape(kOutputTensor)); in execute()
209 context->getOutputShape(kOutputTensor)); in execute()
214 context->getOutputShape(kOutputTensor)); in execute()
DSelect.cpp64 context->getOutputShape(kOutputTensor)); in executeTyped()
99 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DDequantize.cpp101 Shape output = context->getOutputShape(kOutputTensor); in prepare()
108 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DInstanceNormalization.cpp128 context->getOutputShape(kOutputTensor)); in execute()
137 context->getOutputShape(kOutputTensor)); in execute()
DLogicalNot.cpp57 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DTransposeConv2D.cpp511 Shape output = context->getOutputShape(kOutputTensor); in prepare()
523 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
535 context->getOutputShape(kOutputTensor)); in execute()
544 context->getOutputShape(kOutputTensor)); in execute()
557 context->getOutputShape(kOutputTensor)); in execute()
566 context->getOutputShape(kOutputTensor)); in execute()
DPooling.cpp336 context->getOutputShape(kOutputTensor))
340 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAveragePool()
354 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeL2Pool()
367 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMaxPool()
DBidirectionalSequenceRNN.cpp114 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()
118 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
360 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()
367 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare()
DConv2D.cpp540 Shape output = context->getOutputShape(kOutputTensor); in prepare()
552 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
567 context->getOutputShape(kOutputTensor)); in execute()
579 context->getOutputShape(kOutputTensor)); in execute()
595 context->getOutputShape(kOutputTensor)); in execute()
607 context->getOutputShape(kOutputTensor)); in execute()
DRoiPooling.cpp262 context->getOutputShape(kOutputTensor)); in execute()
274 context->getOutputShape(kOutputTensor)); in execute()
286 context->getOutputShape(kOutputTensor)); in execute()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp86 Shape output = context->getOutputShape(kOutputTensor); in prepare()
126 Shape outputShape = context->getOutputShape(kOutputTensor); in execute()

12