Home
last modified time | relevance | path

Searched refs:getInputShape (Results 1 – 25 of 45) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DSlice.cpp106 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
110 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
114 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
143 context->getInputShape(kInputTensor), in execute()
145 context->getInputShape(kBeginTensor), in execute()
147 context->getInputShape(kSizeTensor), in execute()
152 context->getInputShape(kInputTensor), in execute()
154 context->getInputShape(kBeginTensor), in execute()
156 context->getInputShape(kSizeTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
[all …]
DUnidirectionalSequenceLSTM.cpp197 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
205 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
210 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
216 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
222 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
226 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
232 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
238 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
242 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
256 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
DGather.cpp86 Shape input = context->getInputShape(kInputTensor); in prepare()
89 Shape indices = context->getInputShape(kInputIndices); in prepare()
106 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
110 context->getInputShape(kInputTensor), axis, in execute()
112 context->getInputShape(kInputIndices), in execute()
116 context->getInputShape(kInputTensor), axis, in execute()
118 context->getInputShape(kInputIndices), in execute()
122 context->getInputShape(kInputTensor), axis, in execute()
124 context->getInputShape(kInputIndices), in execute()
128 context->getInputShape(kInputTensor), axis, in execute()
[all …]
DQLSTM.cpp178 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
185 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
190 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
196 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
202 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
206 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
212 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
218 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
222 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
236 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
DFullyConnected.cpp249 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
250 const float weightsScale = context->getInputShape(kWeightsTensor).scale; in validate()
282 Shape input = context->getInputShape(kInputTensor); in validate()
283 Shape weights = context->getInputShape(kWeightsTensor); in validate()
284 Shape bias = context->getInputShape(kBiasTensor); in validate()
293 Shape input = context->getInputShape(kInputTensor); in prepare()
294 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
295 Shape bias = context->getInputShape(kBiasTensor); in prepare()
307 context->getInputShape(kInputTensor), in execute()
309 context->getInputShape(kWeightsTensor), in execute()
[all …]
DBroadcast.cpp456 Shape input1 = context->getInputShape(kInputTensor1); in validate()
457 Shape input2 = context->getInputShape(kInputTensor2); in validate()
469 const Shape& input1 = context->getInputShape(kInputTensor1); in validate()
470 const Shape& input2 = context->getInputShape(kInputTensor2); in validate()
480 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
481 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
495 context->getInputShape(kInputTensor1), in executeAdd()
497 context->getInputShape(kInputTensor2), in executeAdd()
503 context->getInputShape(kInputTensor1), in executeAdd()
505 context->getInputShape(kInputTensor2), in executeAdd()
[all …]
DTranspose.cpp90 const Shape& input = context->getInputShape(kInputTensor); in validate()
103 const Shape& input = context->getInputShape(kInputTensor); in prepare()
115 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
143 context->getInputShape(kInputTensor), in execute()
145 context->getInputShape(kPermTensor), in execute()
150 context->getInputShape(kInputTensor), in execute()
152 context->getInputShape(kPermTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
159 context->getInputShape(kPermTensor), in execute()
164 context->getInputShape(kInputTensor), in execute()
[all …]
DBidirectionalSequenceRNN.cpp125 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
128 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
130 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
135 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
137 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
156 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
157 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
158 Shape bwAuxWeightsShape = context->getInputShape(kBwAuxWeightsTensor); in executeTyped()
368 Shape input = context->getInputShape(kInputTensor); in prepare()
369 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
[all …]
DPRelu.cpp120 Shape input = context->getInputShape(kInputTensor); in prepare()
121 Shape alpha = context->getInputShape(kAlphaTensor); in prepare()
136 context->getInputShape(kInputTensor), in execute()
138 context->getInputShape(kAlphaTensor), in execute()
147 context->getInputShape(kInputTensor), in execute()
149 context->getInputShape(kAlphaTensor), in execute()
154 context->getInputShape(kInputTensor), in execute()
156 context->getInputShape(kAlphaTensor), in execute()
162 context->getInputShape(kInputTensor), in execute()
164 context->getInputShape(kAlphaTensor), in execute()
DComparisons.cpp77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
113 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
114 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DActivation.cpp378 const Shape& input = context->getInputShape(kInputTensor); in validate()
400 Shape input = context->getInputShape(kInputTensor); in prepare()
439 context->getInputShape(kInputTensor), in executeRelu()
444 context->getInputShape(kInputTensor), in executeRelu()
449 context->getInputShape(kInputTensor), in executeRelu()
454 context->getInputShape(kInputTensor), in executeRelu()
468 context->getInputShape(kInputTensor), in executeRelu1()
473 context->getInputShape(kInputTensor), in executeRelu1()
478 context->getInputShape(kInputTensor), in executeRelu1()
483 context->getInputShape(kInputTensor), in executeRelu1()
[all …]
DChannelShuffle.cpp72 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
86 Shape input = context->getInputShape(kInputTensor); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
102 context->getInputShape(kInputTensor), numGroups, axis, in execute()
106 context->getInputShape(kInputTensor), numGroups, axis, in execute()
110 context->getInputShape(kInputTensor), numGroups, axis, in execute()
114 context->getInputShape(kInputTensor), numGroups, axis, in execute()
DLogicalAndOr.cpp75 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()
DRoiPooling.cpp231 Shape input = context->getInputShape(kInputTensor); in prepare()
232 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
233 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
280 context->getInputShape(kInputTensor), in execute()
282 context->getInputShape(kRoiTensor), in execute()
284 context->getInputShape(kBatchSplitTensor), in execute()
292 context->getInputShape(kInputTensor), in execute()
294 context->getInputShape(kRoiTensor), in execute()
296 context->getInputShape(kBatchSplitTensor), in execute()
304 context->getInputShape(kInputTensor), in execute()
[all …]
DTransposeConv2D.cpp75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
78 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
79 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4); in initialize()
484 Shape input = context->getInputShape(kInputTensor); in prepare()
485 Shape filter = context->getInputShape(kFilterTensor); in prepare()
486 Shape bias = context->getInputShape(kBiasTensor); in prepare()
549 context->getInputShape(kInputTensor), in execute()
551 context->getInputShape(kFilterTensor), in execute()
553 context->getInputShape(kBiasTensor), param, in execute()
558 context->getInputShape(kInputTensor), in execute()
[all …]
DElementwise.cpp54 context->getInputShape(kInputTensor), in execute()
58 context->getInputShape(kInputTensor), in execute()
72 context->getInputShape(kInputTensor), in executeAbs()
76 context->getInputShape(kInputTensor), in executeAbs()
81 context->getInputShape(kInputTensor), in executeAbs()
124 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
135 Shape input = context->getInputShape(kInputTensor); in prepare()
142 Shape input = context->getInputShape(kInputTensor); in prepareFloor()
DDepthwiseConv2D.cpp94 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
95 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
467 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
468 const float filterScale = context->getInputShape(kFilterTensor).scale; in validate()
469 const float outputScale = context->getInputShape(kOutputTensor).scale; in validate()
512 Shape input = context->getInputShape(kInputTensor); in prepare()
513 Shape filter = context->getInputShape(kFilterTensor); in prepare()
514 Shape bias = context->getInputShape(kBiasTensor); in prepare()
578 context->getInputShape(kInputTensor), in execute()
580 context->getInputShape(kFilterTensor), in execute()
[all …]
DSelect.cpp64 context->getInputShape(kInputCondition), context->getInputBuffer<T>(kInputTensor1), in executeTyped()
65 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
66 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
88 Shape inputCondition = context->getInputShape(kInputCondition); in prepare()
89 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
101 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
DUnidirectionalSequenceRNN.cpp68 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
70 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
72 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()
DGenerateProposals.cpp230 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
231 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
232 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
233 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
279 context->getInputShape(kRoiTensor), in execute()
281 context->getInputShape(kDeltaTensor), in execute()
283 context->getInputShape(kBatchesTensor), in execute()
285 context->getInputShape(kImageInfoTensor), in execute()
291 context->getInputShape(kRoiTensor), in execute()
293 context->getInputShape(kDeltaTensor), in execute()
[all …]
DConv2D.cpp107 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
537 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
538 const auto filterRank = getNumberOfDimensions(context->getInputShape(kFilterTensor)); in validate()
585 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
586 const float filterScale = context->getInputShape(kFilterTensor).scale; in validate()
587 const float outputScale = context->getInputShape(kOutputTensor).scale; in validate()
630 Shape input = context->getInputShape(kInputTensor); in prepare()
631 Shape filter = context->getInputShape(kFilterTensor); in prepare()
632 Shape bias = context->getInputShape(kBiasTensor); in prepare()
[all …]
DNeg.cpp66 Shape input = context->getInputShape(kInputTensor); in prepare()
76 context->getInputShape(kInputTensor), in execute()
80 context->getInputShape(kInputTensor), in execute()
84 context->getInputShape(kInputTensor), in execute()
DRoiAlign.cpp386 Shape input = context->getInputShape(kInputTensor); in prepare()
387 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
388 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
444 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
448 context->getInputShape(kInputTensor), in execute()
450 context->getInputShape(kRoiTensor), in execute()
452 context->getInputShape(kBatchSplitTensor), in execute()
462 context->getInputShape(kInputTensor), in execute()
464 context->getInputShape(kRoiTensor), in execute()
466 context->getInputShape(kBatchSplitTensor), in execute()
[all …]
DConcatenation.cpp102 inputShapes.push_back(context->getInputShape(i)); in concatenation()
114 const auto currentSize = getNumberOfElements(context->getInputShape(i)); in concatenation()
124 inputShapes.push_back(context->getInputShape(i)); in concatenation()
161 const Shape& input = context->getInputShape(i); in validate()
167 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate()
179 const Shape& input0 = context->getInputShape(0); in prepare()
188 const Shape& input = context->getInputShape(i); in prepare()
DHeatmapMaxKeypoint.cpp261 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare()
262 Shape boxesShape = context->getInputShape(kBoxesTensor); in prepare()
304 const auto heatmapShape = context->getInputShape(kHeatmapTensor); in execute()
306 const auto boxesShape = context->getInputShape(kBoxesTensor); in execute()
327 context->getInputShape(kHeatmapTensor), in execute()
329 context->getInputShape(kBoxesTensor), layout, in execute()
339 context->getInputShape(kHeatmapTensor), in execute()
341 context->getInputShape(kBoxesTensor), layout, in execute()
350 context->getInputShape(kHeatmapTensor), in execute()
352 context->getInputShape(kBoxesTensor), layout, in execute()

12