Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DActivation.cpp410 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
435 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
441 context->getOutputShape(kOutputTensor)); in executeRelu()
446 context->getOutputShape(kOutputTensor)); in executeRelu()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
456 context->getOutputShape(kOutputTensor)); in executeRelu()
464 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
470 context->getOutputShape(kOutputTensor)); in executeRelu1()
475 context->getOutputShape(kOutputTensor)); in executeRelu1()
480 context->getOutputShape(kOutputTensor)); in executeRelu1()
[all …]
DQuantize.cpp91 Shape output = context->getOutputShape(kOutputTensor); in prepare()
98 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
106 context->getOutputShape(kOutputTensor)); in execute()
110 context->getOutputShape(kOutputTensor)); in execute()
116 context->getOutputShape(kOutputTensor)); in execute()
120 context->getOutputShape(kOutputTensor)); in execute()
DBroadcast.cpp455 Shape output = context->getOutputShape(kOutputTensor); in validate()
482 Shape output = context->getOutputShape(kOutputTensor); in prepare()
491 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
500 context->getOutputShape(kOutputTensor)); in executeAdd()
508 context->getOutputShape(kOutputTensor)); in executeAdd()
516 context->getOutputShape(kOutputTensor)); in executeAdd()
524 context->getOutputShape(kOutputTensor)); in executeAdd()
532 context->getOutputShape(kOutputTensor), in executeAdd()
541 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
550 context->getOutputShape(kOutputTensor)); in executeMul()
[all …]
DTranspose.cpp105 Shape output = context->getOutputShape(kOutputTensor); in prepare()
138 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
147 context->getOutputShape(kOutputTensor)); in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
161 context->getOutputShape(kOutputTensor)); in execute()
168 context->getOutputShape(kOutputTensor)); in execute()
DSlice.cpp121 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
139 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
149 context->getOutputShape(kOutputTensor)); in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
167 context->getOutputShape(kOutputTensor)); in execute()
176 context->getOutputShape(kOutputTensor)); in execute()
185 context->getOutputShape(kOutputTensor)); in execute()
DGenerateProposals.cpp234 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
275 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
287 context->getOutputShape(kOutputTensor)); in execute()
299 context->getOutputShape(kOutputTensor)); in execute()
312 context->getOutputShape(kOutputTensor)); in execute()
323 context->getOutputShape(kOutputTensor)); in execute()
576 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
580 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
584 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
588 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
[all …]
DHeatmapMaxKeypoint.cpp281 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()
286 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare()
308 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute()
310 const auto outputKeypointShape = context->getOutputShape(kOutputKeypointTensor); in execute()
331 context->getOutputShape(kOutputScoreTensor), in execute()
333 context->getOutputShape(kOutputKeypointTensor), 1e-5f, in execute()
343 context->getOutputShape(kOutputScoreTensor), in execute()
345 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
354 context->getOutputShape(kOutputScoreTensor), in execute()
356 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
DPRelu.cpp123 Shape output = context->getOutputShape(kOutputTensor); in prepare()
140 context->getOutputShape(kOutputTensor)); in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
166 context->getOutputShape(kOutputTensor)); in execute()
DElu.cpp80 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
87 context->getOutputShape(kOutputTensor)); in execute()
93 context->getOutputShape(kOutputTensor)); in execute()
DConcatenation.cpp106 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()
159 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
200 Shape output = context->getOutputShape(kOutputTensor); in prepare()
209 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DComparisons.cpp79 context->getOutputShape(kOutputTensor)); in executeLessTyped()
88 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
97 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
106 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
115 context->getOutputShape(kOutputTensor)); in executeGreaterEqualTyped()
124 context->getOutputShape(kOutputTensor)); in executeGreaterTyped()
151 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DFullyConnected.cpp251 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
296 Shape output = context->getOutputShape(kOutputTensor); in prepare()
303 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
314 context->getOutputShape(kOutputTensor)); in execute()
324 context->getOutputShape(kOutputTensor)); in execute()
334 context->getOutputShape(kOutputTensor)); in execute()
344 context->getOutputShape(kOutputTensor)); in execute()
DLogicalAndOr.cpp77 Shape output = context->getOutputShape(kOutputTensor); in prepare()
87 context->getOutputShape(kOutputTensor)); in executeAnd()
95 context->getOutputShape(kOutputTensor)); in executeOr()
DSoftmax.cpp273 Shape output = context->getOutputShape(kOutputTensor); in prepare()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
290 context->getOutputShape(kOutputTensor)); in execute()
296 context->getOutputShape(kOutputTensor)); in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
308 context->getOutputShape(kOutputTensor)); in execute()
DL2Normalization.cpp241 Shape output = context->getOutputShape(kOutputTensor); in prepare()
267 context->getOutputShape(kOutputTensor)); in execute()
272 context->getOutputShape(kOutputTensor)); in execute()
277 context->getOutputShape(kOutputTensor)); in execute()
282 context->getOutputShape(kOutputTensor)); in execute()
DFill.cpp41 const int numElements = getNumberOfElements(context->getOutputShape(kOutputTensor)); in executeTyped()
90 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
DResizeImageOps.cpp277 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
289 context->getOutputShape(kOutputTensor)); in execute()
294 context->getOutputShape(kOutputTensor)); in execute()
299 context->getOutputShape(kOutputTensor)); in execute()
304 context->getOutputShape(kOutputTensor)); in execute()
DTransposeConv2D.cpp531 Shape output = context->getOutputShape(kOutputTensor); in prepare()
543 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
555 context->getOutputShape(kOutputTensor)); in execute()
564 context->getOutputShape(kOutputTensor)); in execute()
577 context->getOutputShape(kOutputTensor)); in execute()
586 context->getOutputShape(kOutputTensor)); in execute()
602 context->getOutputShape(kOutputTensor)); in execute()
612 context->getOutputShape(kOutputTensor)); in execute()
DSelect.cpp67 context->getOutputShape(kOutputTensor)); in executeTyped()
104 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DDepthwiseConv2D.cpp560 Shape output = context->getOutputShape(kOutputTensor); in prepare()
572 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
588 context->getOutputShape(kOutputTensor)); in execute()
601 context->getOutputShape(kOutputTensor)); in execute()
617 context->getOutputShape(kOutputTensor)); in execute()
630 context->getOutputShape(kOutputTensor)); in execute()
649 context->getOutputShape(kOutputTensor)); in execute()
663 context->getOutputShape(kOutputTensor)); in execute()
DRoiPooling.cpp289 context->getOutputShape(kOutputTensor)); in execute()
301 context->getOutputShape(kOutputTensor)); in execute()
313 context->getOutputShape(kOutputTensor)); in execute()
325 context->getOutputShape(kOutputTensor)); in execute()
DDequantize.cpp110 Shape output = context->getOutputShape(kOutputTensor); in prepare()
117 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DInstanceNormalization.cpp141 context->getOutputShape(kOutputTensor)); in execute()
150 context->getOutputShape(kOutputTensor)); in execute()
DConv2D.cpp682 Shape output = context->getOutputShape(kOutputTensor); in prepare()
694 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
709 context->getOutputShape(kOutputTensor)); in execute()
721 context->getOutputShape(kOutputTensor)); in execute()
737 context->getOutputShape(kOutputTensor)); in execute()
749 context->getOutputShape(kOutputTensor)); in execute()
768 context->getOutputShape(kOutputTensor)); in execute()
781 context->getOutputShape(kOutputTensor)); in execute()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp89 Shape output = context->getOutputShape(kOutputTensor); in prepare()
129 Shape outputShape = context->getOutputShape(kOutputTensor); in execute()

12