Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp184 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
192 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
197 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
203 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
209 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
213 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
219 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
225 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
229 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
243 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
DSlice.cpp94 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
98 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
102 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
131 context->getInputShape(kInputTensor), in execute()
133 context->getInputShape(kBeginTensor), in execute()
135 context->getInputShape(kSizeTensor), in execute()
140 context->getInputShape(kInputTensor), in execute()
142 context->getInputShape(kBeginTensor), in execute()
144 context->getInputShape(kSizeTensor), in execute()
149 context->getInputShape(kInputTensor), in execute()
[all …]
DGather.cpp78 Shape input = context->getInputShape(kInputTensor); in prepare()
81 Shape indices = context->getInputShape(kInputIndices); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
102 context->getInputShape(kInputTensor), axis, in execute()
104 context->getInputShape(kInputIndices), in execute()
108 context->getInputShape(kInputTensor), axis, in execute()
110 context->getInputShape(kInputIndices), in execute()
114 context->getInputShape(kInputTensor), axis, in execute()
116 context->getInputShape(kInputIndices), in execute()
120 context->getInputShape(kInputTensor), axis, in execute()
[all …]
DFullyConnected.cpp167 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
168 const float weightsScale = context->getInputShape(kWeightsTensor).scale; in validate()
194 Shape input = context->getInputShape(kInputTensor); in prepare()
195 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
196 Shape bias = context->getInputShape(kBiasTensor); in prepare()
232 context->getInputShape(kInputTensor), in execute()
234 context->getInputShape(kWeightsTensor), in execute()
236 context->getInputShape(kBiasTensor), in execute()
242 context->getInputShape(kInputTensor), in execute()
244 context->getInputShape(kWeightsTensor), in execute()
[all …]
DComparisons.cpp71 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
72 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
80 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
81 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
89 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
90 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
98 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
99 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
107 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
108 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DBroadcast.cpp368 Shape input1 = context->getInputShape(kInputTensor1); in validate()
369 Shape input2 = context->getInputShape(kInputTensor2); in validate()
383 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
384 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
398 context->getInputShape(kInputTensor1), in executeAdd()
400 context->getInputShape(kInputTensor2), in executeAdd()
406 context->getInputShape(kInputTensor1), in executeAdd()
408 context->getInputShape(kInputTensor2), in executeAdd()
414 context->getInputShape(kInputTensor1), in executeAdd()
416 context->getInputShape(kInputTensor2), in executeAdd()
[all …]
DBidirectionalSequenceRNN.cpp80 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
83 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
85 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
90 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
92 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
105 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
106 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
107 Shape bwAuxWeightsShape = context->getInputShape(kBwAuxWeightsTensor); in executeTyped()
291 Shape input = context->getInputShape(kInputTensor); in prepare()
292 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
[all …]
DTranspose.cpp90 const Shape& input = context->getInputShape(kInputTensor); in prepare()
102 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
130 context->getInputShape(kInputTensor), in execute()
132 context->getInputShape(kPermTensor), in execute()
137 context->getInputShape(kInputTensor), in execute()
139 context->getInputShape(kPermTensor), in execute()
144 context->getInputShape(kInputTensor), in execute()
146 context->getInputShape(kPermTensor), in execute()
DLogicalAndOr.cpp73 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
74 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
83 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
91 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
92 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
DPRelu.cpp109 Shape input = context->getInputShape(kInputTensor); in prepare()
110 Shape alpha = context->getInputShape(kAlphaTensor); in prepare()
125 context->getInputShape(kInputTensor), in execute()
127 context->getInputShape(kAlphaTensor), in execute()
136 context->getInputShape(kInputTensor), in execute()
138 context->getInputShape(kAlphaTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
145 context->getInputShape(kAlphaTensor), in execute()
DActivation.cpp230 Shape input = context->getInputShape(kInputTensor); in prepare()
260 context->getInputShape(kInputTensor), in executeRelu()
265 context->getInputShape(kInputTensor), in executeRelu()
270 context->getInputShape(kInputTensor), in executeRelu()
284 context->getInputShape(kInputTensor), in executeRelu1()
289 context->getInputShape(kInputTensor), in executeRelu1()
294 context->getInputShape(kInputTensor), in executeRelu1()
308 context->getInputShape(kInputTensor), in executeRelu6()
313 context->getInputShape(kInputTensor), in executeRelu6()
318 context->getInputShape(kInputTensor), in executeRelu6()
[all …]
DSelect.cpp61 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()
83 Shape inputCondition = context->getInputShape(kInputCondition); in prepare()
84 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
96 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
DRoiPooling.cpp204 Shape input = context->getInputShape(kInputTensor); in prepare()
205 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
206 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
253 context->getInputShape(kInputTensor), in execute()
255 context->getInputShape(kRoiTensor), in execute()
257 context->getInputShape(kBatchSplitTensor), in execute()
265 context->getInputShape(kInputTensor), in execute()
267 context->getInputShape(kRoiTensor), in execute()
269 context->getInputShape(kBatchSplitTensor), in execute()
277 context->getInputShape(kInputTensor), in execute()
[all …]
DUnidirectionalSequenceRNN.cpp59 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
61 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
63 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped()
132 Shape input = context->getInputShape(kInputTensor); in prepare()
133 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
134 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare()
135 Shape bias = context->getInputShape(kBiasTensor); in prepare()
136 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare()
DNeg.cpp64 Shape input = context->getInputShape(kInputTensor); in prepare()
74 context->getInputShape(kInputTensor), in execute()
78 context->getInputShape(kInputTensor), in execute()
82 context->getInputShape(kInputTensor), in execute()
DChannelShuffle.cpp75 Shape input = context->getInputShape(kInputTensor); in prepare()
87 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
91 context->getInputShape(kInputTensor), numGroups, axis, in execute()
95 context->getInputShape(kInputTensor), numGroups, axis, in execute()
99 context->getInputShape(kInputTensor), numGroups, axis, in execute()
DConv2D.cpp94 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
95 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
448 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
449 const float filterScale = context->getInputShape(kFilterTensor).scale; in validate()
450 const float outputScale = context->getInputShape(kOutputTensor).scale; in validate()
490 Shape input = context->getInputShape(kInputTensor); in prepare()
491 Shape filter = context->getInputShape(kFilterTensor); in prepare()
492 Shape bias = context->getInputShape(kBiasTensor); in prepare()
558 context->getInputShape(kInputTensor), in execute()
560 context->getInputShape(kFilterTensor), in execute()
[all …]
DGenerateProposals.cpp194 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
195 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
196 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
197 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
240 context->getInputShape(kRoiTensor), in execute()
242 context->getInputShape(kDeltaTensor), in execute()
244 context->getInputShape(kBatchesTensor), in execute()
246 context->getInputShape(kImageInfoTensor), in execute()
252 context->getInputShape(kRoiTensor), in execute()
254 context->getInputShape(kDeltaTensor), in execute()
[all …]
DTransposeConv2D.cpp65 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
68 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
69 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4); in initialize()
466 Shape input = context->getInputShape(kInputTensor); in prepare()
467 Shape filter = context->getInputShape(kFilterTensor); in prepare()
468 Shape bias = context->getInputShape(kBiasTensor); in prepare()
529 context->getInputShape(kInputTensor), in execute()
531 context->getInputShape(kFilterTensor), in execute()
533 context->getInputShape(kBiasTensor), param, in execute()
538 context->getInputShape(kInputTensor), in execute()
[all …]
DRoiAlign.cpp369 Shape input = context->getInputShape(kInputTensor); in prepare()
370 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
371 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
427 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
431 context->getInputShape(kInputTensor), in execute()
433 context->getInputShape(kRoiTensor), in execute()
435 context->getInputShape(kBatchSplitTensor), in execute()
445 context->getInputShape(kInputTensor), in execute()
447 context->getInputShape(kRoiTensor), in execute()
449 context->getInputShape(kBatchSplitTensor), in execute()
[all …]
DLogSoftmax.cpp94 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
99 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
103 context->getInputShape(kInputTensor), in execute()
108 context->getInputShape(kInputTensor), in execute()
DHeatmapMaxKeypoint.cpp226 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare()
227 Shape boxesShape = context->getInputShape(kBoxesTensor); in prepare()
264 const auto heatmapShape = context->getInputShape(kHeatmapTensor); in execute()
266 const auto boxesShape = context->getInputShape(kBoxesTensor); in execute()
287 context->getInputShape(kHeatmapTensor), in execute()
289 context->getInputShape(kBoxesTensor), layout, in execute()
299 context->getInputShape(kHeatmapTensor), in execute()
301 context->getInputShape(kBoxesTensor), layout, in execute()
DL2Normalization.cpp162 } else if (context->getInputShape(kInputTensor).dimensions.size() != 4) { in validate()
170 const Shape& input = context->getInputShape(kInputTensor); in prepare()
194 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
198 context->getInputShape(kInputTensor), axis, in execute()
203 context->getInputShape(kInputTensor), axis, in execute()
208 context->getInputShape(kInputTensor), axis, in execute()
DElementwise.cpp51 context->getInputShape(kInputTensor), in execute()
55 context->getInputShape(kInputTensor), in execute()
77 Shape input = context->getInputShape(kInputTensor); in prepare()
DLogicalNot.cpp56 Shape input = context->getInputShape(kInputTensor); in prepare()
64 context->getInputShape(kInputTensor), in execute()

12