| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Slice.cpp | 111 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
115 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
119 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
148 context->getInputShape(kInputTensor), in execute()
150 context->getInputShape(kBeginTensor), in execute()
152 context->getInputShape(kSizeTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
161 context->getInputShape(kSizeTensor), in execute()
170 context->getInputShape(kSizeTensor), in execute()
179 context->getInputShape(kSizeTensor), in execute()
[all …]
|
| H A D | UnidirectionalSequenceLSTM.cpp | 200 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
300 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
324 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()
377 Shape outputShape = context->getInputShape(kInputTensor); in prepare()
390 Shape cellStateOutTensor = context->getInputShape(kCellStateInTensor); in prepare()
427 context->getInputShape(kInputTensor), in execute()
432 context->getInputShape(kInputToOutputWeightsTensor), in execute()
437 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
480 context->getInputShape(kInputTensor), in execute()
485 context->getInputShape(kInputToOutputWeightsTensor), in execute()
[all …]
|
| H A D | QLSTM.cpp | 179 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
278 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
298 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
302 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
353 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
357 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
369 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
372 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
377 context->getInputShape(kRecurrentToForgetWeightsTensor); in execute()
385 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
[all …]
|
| H A D | Broadcast.cpp | 460 Shape input1 = context->getInputShape(kInputTensor1); in validate()
461 Shape input2 = context->getInputShape(kInputTensor2); in validate()
473 const Shape& input1 = context->getInputShape(kInputTensor1); in validate()
474 const Shape& input2 = context->getInputShape(kInputTensor2); in validate()
486 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
487 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
517 context->getInputShape(kInputTensor1), in executeAdd()
519 context->getInputShape(kInputTensor2), in executeAdd()
525 context->getInputShape(kInputTensor1), in executeAdd()
527 context->getInputShape(kInputTensor2), in executeAdd()
[all …]
|
| H A D | Gather.cpp | 83 Shape input = context->getInputShape(kInputTensor); in prepare()
86 Shape indices = context->getInputShape(kInputIndices); in prepare()
107 context->getInputShape(kInputTensor), axis, in execute()
109 context->getInputShape(kInputIndices), in execute()
113 context->getInputShape(kInputTensor), axis, in execute()
115 context->getInputShape(kInputIndices), in execute()
119 context->getInputShape(kInputTensor), axis, in execute()
121 context->getInputShape(kInputIndices), in execute()
125 context->getInputShape(kInputTensor), axis, in execute()
127 context->getInputShape(kInputIndices), in execute()
[all …]
|
| H A D | FullyConnected.cpp | 252 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
284 Shape input = context->getInputShape(kInputTensor); in validate()
285 Shape weights = context->getInputShape(kWeightsTensor); in validate()
286 Shape bias = context->getInputShape(kBiasTensor); in validate()
296 Shape input = context->getInputShape(kInputTensor); in prepare()
297 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
298 Shape bias = context->getInputShape(kBiasTensor); in prepare()
314 context->getInputShape(kBiasTensor), in execute()
324 context->getInputShape(kBiasTensor), in execute()
334 context->getInputShape(kBiasTensor), in execute()
[all …]
|
| H A D | Transpose.cpp | 92 const Shape& input = context->getInputShape(kInputTensor); in validate()
107 const Shape& input = context->getInputShape(kInputTensor); in prepare()
119 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
147 context->getInputShape(kInputTensor), in execute()
149 context->getInputShape(kPermTensor), in execute()
154 context->getInputShape(kInputTensor), in execute()
156 context->getInputShape(kPermTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
163 context->getInputShape(kPermTensor), in execute()
168 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A D | BidirectionalSequenceRNN.cpp | 123 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
126 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
133 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
154 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
367 Shape input = context->getInputShape(kInputTensor); in prepare()
368 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
370 Shape fwBias = context->getInputShape(kFwBiasTensor); in prepare()
372 Shape bwWeights = context->getInputShape(kBwWeightsTensor); in prepare()
374 Shape bwBias = context->getInputShape(kBwBiasTensor); in prepare()
377 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
[all …]
|
| H A D | PRelu.cpp | 122 Shape input = context->getInputShape(kInputTensor); in prepare()
123 Shape alpha = context->getInputShape(kAlphaTensor); in prepare()
138 context->getInputShape(kInputTensor), in execute()
140 context->getInputShape(kAlphaTensor), in execute()
149 context->getInputShape(kInputTensor), in execute()
151 context->getInputShape(kAlphaTensor), in execute()
156 context->getInputShape(kInputTensor), in execute()
158 context->getInputShape(kAlphaTensor), in execute()
164 context->getInputShape(kInputTensor), in execute()
166 context->getInputShape(kAlphaTensor), in execute()
|
| H A D | Activation.cpp | 382 const Shape& input = context->getInputShape(kInputTensor); in validate()
410 Shape input = context->getInputShape(kInputTensor); in prepare()
449 context->getInputShape(kInputTensor), in executeRelu()
454 context->getInputShape(kInputTensor), in executeRelu()
459 context->getInputShape(kInputTensor), in executeRelu()
478 context->getInputShape(kInputTensor), in executeRelu1()
483 context->getInputShape(kInputTensor), in executeRelu1()
488 context->getInputShape(kInputTensor), in executeRelu1()
507 context->getInputShape(kInputTensor), in executeRelu6()
512 context->getInputShape(kInputTensor), in executeRelu6()
[all …]
|
| H A D | Comparisons.cpp | 74 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
75 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
83 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
92 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
93 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
102 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
111 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
120 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterTyped()
146 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
[all …]
|
| H A D | RoiPooling.cpp | 232 Shape input = context->getInputShape(kInputTensor); in prepare()
233 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
234 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
281 context->getInputShape(kInputTensor), in execute()
283 context->getInputShape(kRoiTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
295 context->getInputShape(kRoiTensor), in execute()
305 context->getInputShape(kInputTensor), in execute()
307 context->getInputShape(kRoiTensor), in execute()
317 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A D | TransposeConv2D.cpp | 75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
489 Shape input = context->getInputShape(kInputTensor); in prepare()
490 Shape filter = context->getInputShape(kFilterTensor); in prepare()
491 Shape bias = context->getInputShape(kBiasTensor); in prepare()
554 context->getInputShape(kInputTensor), in execute()
575 context->getInputShape(kInputTensor), in execute()
577 context->getInputShape(kFilterTensor), in execute()
582 context->getInputShape(kBiasTensor), param, in execute()
602 context->getInputShape(kInputTensor), in execute()
604 context->getInputShape(kFilterTensor), in execute()
[all …]
|
| H A D | ChannelShuffle.cpp | 69 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
83 Shape input = context->getInputShape(kInputTensor); in prepare()
95 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
99 context->getInputShape(kInputTensor), numGroups, axis, in execute()
103 context->getInputShape(kInputTensor), numGroups, axis, in execute()
107 context->getInputShape(kInputTensor), numGroups, axis, in execute()
111 context->getInputShape(kInputTensor), numGroups, axis, in execute()
|
| H A D | DepthwiseConv2D.cpp | 96 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
97 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
520 Shape input = context->getInputShape(kInputTensor); in prepare()
521 Shape filter = context->getInputShape(kFilterTensor); in prepare()
522 Shape bias = context->getInputShape(kBiasTensor); in prepare()
586 context->getInputShape(kInputTensor), in execute()
599 context->getInputShape(kInputTensor), in execute()
615 context->getInputShape(kInputTensor), in execute()
617 context->getInputShape(kFilterTensor), in execute()
649 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A D | LogicalAndOr.cpp | 75 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
76 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
85 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
86 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
93 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
94 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
|
| H A D | GenerateProposals.cpp | 231 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
761 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
821 context->getInputShape(kScoreTensor), in execute()
823 context->getInputShape(kRoiTensor), in execute()
825 context->getInputShape(kBatchesTensor), in execute()
1698 context->getInputShape(kScoreTensor), in execute()
1700 context->getInputShape(kDeltaTensor), in execute()
1702 context->getInputShape(kAnchorTensor), in execute()
1726 context->getInputShape(kScoreTensor), in execute()
1728 context->getInputShape(kDeltaTensor), in execute()
[all …]
|
| H A D | Conv2D.cpp | 637 Shape input = context->getInputShape(kInputTensor); in prepare()
638 Shape filter = context->getInputShape(kFilterTensor); in prepare()
639 Shape bias = context->getInputShape(kBiasTensor); in prepare()
707 context->getInputShape(kInputTensor), in execute()
709 context->getInputShape(kFilterTensor), in execute()
719 context->getInputShape(kInputTensor), in execute()
721 context->getInputShape(kFilterTensor), in execute()
734 context->getInputShape(kInputTensor), in execute()
736 context->getInputShape(kFilterTensor), in execute()
767 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A D | Elementwise.cpp | 51 context->getInputShape(kInputTensor), in execute()
55 context->getInputShape(kInputTensor), in execute()
69 context->getInputShape(kInputTensor), in executeAbs()
73 context->getInputShape(kInputTensor), in executeAbs()
78 context->getInputShape(kInputTensor), in executeAbs()
120 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
129 Shape input = context->getInputShape(kInputTensor); in prepare()
136 Shape input = context->getInputShape(kInputTensor); in prepareFloor()
|
| H A D | UnidirectionalSequenceRNN.cpp | 67 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
69 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
71 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped()
154 Shape input = context->getInputShape(kInputTensor); in prepare()
155 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
156 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare()
157 Shape bias = context->getInputShape(kBiasTensor); in prepare()
158 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare()
190 Shape outputStateShape = context->getInputShape(kHiddenStateTensor); in prepare()
|
| H A D | Select.cpp | 61 context->getInputShape(kInputCondition), context->getInputBuffer<T>(kInputTensor1), in executeTyped()
62 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
63 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
85 Shape inputCondition = context->getInputShape(kInputCondition); in prepare()
86 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
98 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
|
| H A D | RoiAlign.cpp | 389 Shape input = context->getInputShape(kInputTensor); in prepare()
390 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
391 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
451 context->getInputShape(kInputTensor), in execute()
453 context->getInputShape(kRoiTensor), in execute()
465 context->getInputShape(kInputTensor), in execute()
467 context->getInputShape(kRoiTensor), in execute()
479 context->getInputShape(kInputTensor), in execute()
481 context->getInputShape(kRoiTensor), in execute()
493 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A D | Neg.cpp | 63 Shape input = context->getInputShape(kInputTensor); in prepare()
73 context->getInputShape(kInputTensor), in execute()
77 context->getInputShape(kInputTensor), in execute()
81 context->getInputShape(kInputTensor), in execute()
|
| H A D | Concatenation.cpp | 104 inputShapes.push_back(context->getInputShape(i)); in concatenation()
116 const auto currentSize = getNumberOfElements(context->getInputShape(i));
126 inputShapes.push_back(context->getInputShape(i));
164 const Shape& input = context->getInputShape(i); in validate()
171 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate()
185 const Shape& input0 = context->getInputShape(0); in prepare()
194 const Shape& input = context->getInputShape(i); in prepare()
|
| H A D | HeatmapMaxKeypoint.cpp | 263 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare()
264 Shape boxesShape = context->getInputShape(kBoxesTensor); in prepare()
306 const auto heatmapShape = context->getInputShape(kHeatmapTensor); in execute()
308 const auto boxesShape = context->getInputShape(kBoxesTensor); in execute()
329 context->getInputShape(kHeatmapTensor), in execute()
331 context->getInputShape(kBoxesTensor), layout, in execute()
341 context->getInputShape(kHeatmapTensor), in execute()
343 context->getInputShape(kBoxesTensor), layout, in execute()
352 context->getInputShape(kHeatmapTensor), in execute()
354 context->getInputShape(kBoxesTensor), layout, in execute()
|