Home
last modified time | relevance | path

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

123

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp90 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
99 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
104 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
110 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
116 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
120 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
126 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
132 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
136 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
150 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
DSlice.cpp78 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
82 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
86 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
116 context->getInputShape(kInputTensor), in execute()
118 context->getInputShape(kBeginTensor), in execute()
120 context->getInputShape(kSizeTensor), in execute()
125 context->getInputShape(kInputTensor), in execute()
127 context->getInputShape(kBeginTensor), in execute()
129 context->getInputShape(kSizeTensor), in execute()
134 context->getInputShape(kInputTensor), in execute()
[all …]
DQLSTM.cpp63 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
70 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
75 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
81 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
87 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
91 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
97 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
103 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
107 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
121 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
[all …]
DGather.cpp54 Shape input = context->getInputShape(kInputTensor); in prepare()
57 Shape indices = context->getInputShape(kInputIndices); in prepare()
74 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
78 context->getInputShape(kInputTensor), axis, in execute()
80 context->getInputShape(kInputIndices), in execute()
84 context->getInputShape(kInputTensor), axis, in execute()
86 context->getInputShape(kInputIndices), in execute()
90 context->getInputShape(kInputTensor), axis, in execute()
92 context->getInputShape(kInputIndices), in execute()
96 context->getInputShape(kInputTensor), axis, in execute()
[all …]
DBroadcast.cpp442 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
443 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
457 context->getInputShape(kInputTensor1), in executeAdd()
459 context->getInputShape(kInputTensor2), in executeAdd()
465 context->getInputShape(kInputTensor1), in executeAdd()
467 context->getInputShape(kInputTensor2), in executeAdd()
473 context->getInputShape(kInputTensor1), in executeAdd()
475 context->getInputShape(kInputTensor2), in executeAdd()
481 context->getInputShape(kInputTensor1), in executeAdd()
483 context->getInputShape(kInputTensor2), in executeAdd()
[all …]
DFullyConnected.cpp183 Shape input = context->getInputShape(kInputTensor); in prepare()
184 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
185 Shape bias = context->getInputShape(kBiasTensor); in prepare()
197 context->getInputShape(kInputTensor), in execute()
199 context->getInputShape(kWeightsTensor), in execute()
201 context->getInputShape(kBiasTensor), in execute()
207 context->getInputShape(kInputTensor), in execute()
209 context->getInputShape(kWeightsTensor), in execute()
211 context->getInputShape(kBiasTensor), in execute()
217 context->getInputShape(kInputTensor), in execute()
[all …]
DElementwise.cpp77 context->getInputShape(kInputTensor), in execute()
81 context->getInputShape(kInputTensor), in execute()
95 context->getInputShape(kInputTensor), in executeAbs()
99 context->getInputShape(kInputTensor), in executeAbs()
104 context->getInputShape(kInputTensor), in executeAbs()
117 context->getInputShape(kInputTensor), in executeRsqrt()
121 context->getInputShape(kInputTensor), in executeRsqrt()
124 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt()
130 context->getInputShape(kInputTensor), in executeRsqrt()
134 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt()
[all …]
DBidirectionalSequenceRNN.cpp94 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
97 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
99 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
104 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
106 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
125 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
126 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
127 Shape bwAuxWeightsShape = context->getInputShape(kBwAuxWeightsTensor); in executeTyped()
308 Shape input = context->getInputShape(kInputTensor); in prepare()
309 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
[all …]
DTranspose.cpp79 const Shape& input = context->getInputShape(kInputTensor); in prepare()
91 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
119 context->getInputShape(kInputTensor), in execute()
121 context->getInputShape(kPermTensor), in execute()
126 context->getInputShape(kInputTensor), in execute()
128 context->getInputShape(kPermTensor), in execute()
133 context->getInputShape(kInputTensor), in execute()
135 context->getInputShape(kPermTensor), in execute()
140 context->getInputShape(kInputTensor), in execute()
142 context->getInputShape(kPermTensor), in execute()
DComparisons.cpp68 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DActivation.cpp364 Shape input = context->getInputShape(kInputTensor); in prepare()
403 context->getInputShape(kInputTensor), in executeRelu()
408 context->getInputShape(kInputTensor), in executeRelu()
413 context->getInputShape(kInputTensor), in executeRelu()
418 context->getInputShape(kInputTensor), in executeRelu()
432 context->getInputShape(kInputTensor), in executeRelu1()
437 context->getInputShape(kInputTensor), in executeRelu1()
442 context->getInputShape(kInputTensor), in executeRelu1()
447 context->getInputShape(kInputTensor), in executeRelu1()
461 context->getInputShape(kInputTensor), in executeRelu6()
[all …]
DPRelu.cpp101 Shape input = context->getInputShape(kInputTensor); in prepare()
102 Shape alpha = context->getInputShape(kAlphaTensor); in prepare()
117 context->getInputShape(kInputTensor), in execute()
119 context->getInputShape(kAlphaTensor), in execute()
128 context->getInputShape(kInputTensor), in execute()
130 context->getInputShape(kAlphaTensor), in execute()
135 context->getInputShape(kInputTensor), in execute()
137 context->getInputShape(kAlphaTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
145 context->getInputShape(kAlphaTensor), in execute()
DGenerateProposals.cpp199 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
200 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
201 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
202 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
248 context->getInputShape(kRoiTensor), in execute()
250 context->getInputShape(kDeltaTensor), in execute()
252 context->getInputShape(kBatchesTensor), in execute()
254 context->getInputShape(kImageInfoTensor), in execute()
260 context->getInputShape(kRoiTensor), in execute()
262 context->getInputShape(kDeltaTensor), in execute()
[all …]
DRoiPooling.cpp180 Shape input = context->getInputShape(kInputTensor); in prepare()
181 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
182 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
229 context->getInputShape(kInputTensor), in execute()
231 context->getInputShape(kRoiTensor), in execute()
233 context->getInputShape(kBatchSplitTensor), in execute()
241 context->getInputShape(kInputTensor), in execute()
243 context->getInputShape(kRoiTensor), in execute()
245 context->getInputShape(kBatchSplitTensor), in execute()
253 context->getInputShape(kInputTensor), in execute()
[all …]
DTransposeConv2D.cpp70 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
73 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1u); in initialize()
74 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4u); in initialize()
437 Shape input = context->getInputShape(kInputTensor); in prepare()
438 Shape filter = context->getInputShape(kFilterTensor); in prepare()
439 Shape bias = context->getInputShape(kBiasTensor); in prepare()
502 context->getInputShape(kInputTensor), in execute()
504 context->getInputShape(kFilterTensor), in execute()
506 context->getInputShape(kBiasTensor), param, in execute()
511 context->getInputShape(kInputTensor), in execute()
[all …]
DLogicalAndOr.cpp58 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
59 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
68 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
76 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
77 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
DUnidirectionalSequenceRNN.cpp55 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
57 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
59 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped()
119 Shape input = context->getInputShape(kInputTensor); in prepare()
120 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
121 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare()
122 Shape bias = context->getInputShape(kBiasTensor); in prepare()
123 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare()
155 Shape outputStateShape = context->getInputShape(kHiddenStateTensor); in prepare()
DDepthwiseConv2D.cpp92 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
93 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
417 Shape input = context->getInputShape(kInputTensor); in prepare()
418 Shape filter = context->getInputShape(kFilterTensor); in prepare()
419 Shape bias = context->getInputShape(kBiasTensor); in prepare()
483 context->getInputShape(kInputTensor), in execute()
485 context->getInputShape(kFilterTensor), in execute()
487 context->getInputShape(kBiasTensor), param.padding_left, in execute()
496 context->getInputShape(kInputTensor), in execute()
498 context->getInputShape(kFilterTensor), in execute()
[all …]
DSelect.cpp54 context->getInputShape(kInputCondition), context->getInputBuffer<T>(kInputTensor1), in executeTyped()
55 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
56 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
63 Shape inputCondition = context->getInputShape(kInputCondition); in prepare()
64 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
76 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
DBatchMatmul.cpp203 Shape inputLHSTensorShape = context->getInputShape(kInputLHSTensor); in prepare()
204 Shape inputRHSTensorShape = context->getInputShape(kInputRHSTensor); in prepare()
233 context->getInputShape(kInputLHSTensor), in execute()
235 context->getInputShape(kInputRHSTensor), in execute()
242 context->getInputShape(kInputLHSTensor), in execute()
244 context->getInputShape(kInputRHSTensor), in execute()
251 context->getInputShape(kInputLHSTensor), in execute()
253 context->getInputShape(kInputRHSTensor), in execute()
260 context->getInputShape(kInputLHSTensor), in execute()
262 context->getInputShape(kInputRHSTensor), in execute()
DChannelShuffle.cpp53 Shape input = context->getInputShape(kInputTensor); in prepare()
65 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
69 context->getInputShape(kInputTensor), numGroups, axis, in execute()
73 context->getInputShape(kInputTensor), numGroups, axis, in execute()
77 context->getInputShape(kInputTensor), numGroups, axis, in execute()
81 context->getInputShape(kInputTensor), numGroups, axis, in execute()
DRoiAlign.cpp334 Shape input = context->getInputShape(kInputTensor); in prepare()
335 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
336 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
392 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
396 context->getInputShape(kInputTensor), in execute()
398 context->getInputShape(kRoiTensor), in execute()
400 context->getInputShape(kBatchSplitTensor), in execute()
410 context->getInputShape(kInputTensor), in execute()
412 context->getInputShape(kRoiTensor), in execute()
414 context->getInputShape(kBatchSplitTensor), in execute()
[all …]
DDensify.cpp148 inputShapes.push_back(context->getInputShape(kInputTensor)); in densify()
150 inputShapes.push_back(context->getInputShape(kInputTravOrder)); in densify()
152 inputShapes.push_back(context->getInputShape(kInputBlockMap)); in densify()
154 inputShapes.push_back(context->getInputShape(kInputDimFormat)); in densify()
156 inputShapes.push_back(context->getInputShape(kInputDimensions)); in densify()
160 inputShapes.push_back(context->getInputShape(i)); in densify()
221 Shape destShape = context->getInputShape(kInputTensor); in prepare()
226 Shape dimensionsShape = context->getInputShape(kInputDimensions); in prepare()
227 Shape blockMapShape = context->getInputShape(kInputBlockMap); in prepare()
DConv2D.cpp105 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
106 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
531 Shape input = context->getInputShape(kInputTensor); in prepare()
532 Shape filter = context->getInputShape(kFilterTensor); in prepare()
533 Shape bias = context->getInputShape(kBiasTensor); in prepare()
601 context->getInputShape(kInputTensor), in execute()
603 context->getInputShape(kFilterTensor), in execute()
605 context->getInputShape(kBiasTensor), param.padding_left, in execute()
613 context->getInputShape(kInputTensor), in execute()
615 context->getInputShape(kFilterTensor), in execute()
[all …]
DNeg.cpp44 Shape input = context->getInputShape(kInputTensor); in prepare()
54 context->getInputShape(kInputTensor), in execute()
58 context->getInputShape(kInputTensor), in execute()
62 context->getInputShape(kInputTensor), in execute()

123