Home
last modified time | relevance | path

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

12

/packages/modules/NeuralNetworks/common/operations/
DActivation.cpp420 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
445 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
456 context->getOutputShape(kOutputTensor)); in executeRelu()
461 context->getOutputShape(kOutputTensor)); in executeRelu()
466 context->getOutputShape(kOutputTensor)); in executeRelu()
474 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
480 context->getOutputShape(kOutputTensor)); in executeRelu1()
485 context->getOutputShape(kOutputTensor)); in executeRelu1()
490 context->getOutputShape(kOutputTensor)); in executeRelu1()
[all …]
DBroadcast.cpp459 Shape output = context->getOutputShape(kOutputTensor); in validate()
488 Shape output = context->getOutputShape(kOutputTensor); in prepare()
497 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
506 context->getOutputShape(kOutputTensor)); in executeAdd()
514 context->getOutputShape(kOutputTensor)); in executeAdd()
522 context->getOutputShape(kOutputTensor)); in executeAdd()
530 context->getOutputShape(kOutputTensor)); in executeAdd()
538 context->getOutputShape(kOutputTensor), in executeAdd()
547 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
556 context->getOutputShape(kOutputTensor)); in executeMul()
[all …]
DGenerateProposals.cpp235 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
276 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
288 context->getOutputShape(kOutputTensor)); in execute()
300 context->getOutputShape(kOutputTensor)); in execute()
313 context->getOutputShape(kOutputTensor)); in execute()
324 context->getOutputShape(kOutputTensor)); in execute()
579 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
583 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
587 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
591 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
[all …]
DQuantize.cpp88 Shape output = context->getOutputShape(kOutputTensor); in prepare()
95 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
103 context->getOutputShape(kOutputTensor)); in execute()
107 context->getOutputShape(kOutputTensor)); in execute()
113 context->getOutputShape(kOutputTensor)); in execute()
117 context->getOutputShape(kOutputTensor)); in execute()
DSlice.cpp126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
144 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
163 context->getOutputShape(kOutputTensor)); in execute()
172 context->getOutputShape(kOutputTensor)); in execute()
181 context->getOutputShape(kOutputTensor)); in execute()
190 context->getOutputShape(kOutputTensor)); in execute()
DHeatmapMaxKeypoint.cpp283 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()
288 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare()
310 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute()
312 const auto outputKeypointShape = context->getOutputShape(kOutputKeypointTensor); in execute()
333 context->getOutputShape(kOutputScoreTensor), in execute()
335 context->getOutputShape(kOutputKeypointTensor), 1e-5f, in execute()
345 context->getOutputShape(kOutputScoreTensor), in execute()
347 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
356 context->getOutputShape(kOutputScoreTensor), in execute()
358 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
DTranspose.cpp109 Shape output = context->getOutputShape(kOutputTensor); in prepare()
142 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
165 context->getOutputShape(kOutputTensor)); in execute()
172 context->getOutputShape(kOutputTensor)); in execute()
DComparisons.cpp76 context->getOutputShape(kOutputTensor)); in executeLessTyped()
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()
DConcatenation.cpp108 context->getOutputShape(kOutputTensor)); in concatenation()
130 std::vector<uint8_t> output_uint8(getNumberOfElements(context->getOutputShape(kOutputTensor))); in concatenation()
131 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation()
162 const Shape& output = context->getOutputShape(kOutputTensor); in validate()
206 Shape output = context->getOutputShape(kOutputTensor); in prepare()
215 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DFullyConnected.cpp254 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
299 Shape output = context->getOutputShape(kOutputTensor); in prepare()
306 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
317 context->getOutputShape(kOutputTensor)); in execute()
327 context->getOutputShape(kOutputTensor)); in execute()
337 context->getOutputShape(kOutputTensor)); in execute()
347 context->getOutputShape(kOutputTensor)); in execute()
DPRelu.cpp125 Shape output = context->getOutputShape(kOutputTensor); in prepare()
142 context->getOutputShape(kOutputTensor)); in execute()
153 context->getOutputShape(kOutputTensor)); in execute()
160 context->getOutputShape(kOutputTensor)); in execute()
168 context->getOutputShape(kOutputTensor)); in execute()
DSoftmax.cpp279 Shape output = context->getOutputShape(kOutputTensor); in prepare()
286 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
296 context->getOutputShape(kOutputTensor)); in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
308 context->getOutputShape(kOutputTensor)); in execute()
314 context->getOutputShape(kOutputTensor)); in execute()
DElu.cpp79 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
86 context->getOutputShape(kOutputTensor)); in execute()
92 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()
DL2Normalization.cpp246 Shape output = context->getOutputShape(kOutputTensor); in prepare()
272 context->getOutputShape(kOutputTensor)); in execute()
277 context->getOutputShape(kOutputTensor)); in execute()
282 context->getOutputShape(kOutputTensor)); in execute()
287 context->getOutputShape(kOutputTensor)); in execute()
DTransposeConv2D.cpp536 Shape output = context->getOutputShape(kOutputTensor); in prepare()
548 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
560 context->getOutputShape(kOutputTensor)); in execute()
569 context->getOutputShape(kOutputTensor)); in execute()
584 context->getOutputShape(kOutputTensor)); in execute()
593 context->getOutputShape(kOutputTensor)); in execute()
611 context->getOutputShape(kOutputTensor)); in execute()
621 context->getOutputShape(kOutputTensor)); in execute()
DResizeImageOps.cpp283 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
295 context->getOutputShape(kOutputTensor)); in execute()
300 context->getOutputShape(kOutputTensor)); in execute()
305 context->getOutputShape(kOutputTensor)); in execute()
310 context->getOutputShape(kOutputTensor)); in execute()
DDepthwiseConv2D.cpp568 Shape output = context->getOutputShape(kOutputTensor); in prepare()
580 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
596 context->getOutputShape(kOutputTensor)); in execute()
609 context->getOutputShape(kOutputTensor)); in execute()
627 context->getOutputShape(kOutputTensor)); in execute()
640 context->getOutputShape(kOutputTensor)); in execute()
661 context->getOutputShape(kOutputTensor)); in execute()
675 context->getOutputShape(kOutputTensor)); in execute()
DConv2D.cpp689 Shape output = context->getOutputShape(kOutputTensor); in prepare()
701 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
716 context->getOutputShape(kOutputTensor)); in execute()
728 context->getOutputShape(kOutputTensor)); in execute()
746 context->getOutputShape(kOutputTensor)); in execute()
758 context->getOutputShape(kOutputTensor)); in execute()
779 context->getOutputShape(kOutputTensor)); in execute()
792 context->getOutputShape(kOutputTensor)); in execute()
DRoiPooling.cpp290 context->getOutputShape(kOutputTensor)); in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
314 context->getOutputShape(kOutputTensor)); in execute()
326 context->getOutputShape(kOutputTensor)); in execute()
DFill.cpp38 const int numElements = getNumberOfElements(context->getOutputShape(kOutputTensor)); in executeTyped()
87 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
DRoiAlign.cpp433 Shape output = context->getOutputShape(kOutputTensor); in prepare()
462 context->getOutputShape(kOutputTensor)); in execute()
476 context->getOutputShape(kOutputTensor)); in execute()
490 context->getOutputShape(kOutputTensor)); in execute()
504 context->getOutputShape(kOutputTensor)); in execute()
DSelect.cpp64 context->getOutputShape(kOutputTensor)); in executeTyped()
101 Shape output = context->getOutputShape(kOutputTensor); in prepare()
DPooling.cpp396 context->getOutputShape(kOutputTensor))
400 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAveragePool()
415 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeL2Pool()
428 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMaxPool()
DBidirectionalSequenceRNN.cpp163 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()
167 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
438 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()
445 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare()

12