/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | UnidirectionalSequenceLSTM.cpp | 90 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 …]
|
D | Slice.cpp | 78 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 …]
|
D | QLSTM.cpp | 63 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 …]
|
D | Gather.cpp | 54 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 …]
|
D | Broadcast.cpp | 442 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 …]
|
D | FullyConnected.cpp | 183 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 …]
|
D | Elementwise.cpp | 77 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 …]
|
D | BidirectionalSequenceRNN.cpp | 94 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 …]
|
D | Transpose.cpp | 79 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()
|
D | Comparisons.cpp | 68 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 …]
|
D | Activation.cpp | 364 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 …]
|
D | PRelu.cpp | 101 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()
|
D | GenerateProposals.cpp | 199 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 …]
|
D | RoiPooling.cpp | 180 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 …]
|
D | TransposeConv2D.cpp | 70 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 …]
|
D | LogicalAndOr.cpp | 58 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()
|
D | UnidirectionalSequenceRNN.cpp | 55 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()
|
D | DepthwiseConv2D.cpp | 92 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 …]
|
D | Select.cpp | 54 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()
|
D | BatchMatmul.cpp | 203 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()
|
D | ChannelShuffle.cpp | 53 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()
|
D | RoiAlign.cpp | 334 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 …]
|
D | Densify.cpp | 148 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()
|
D | Conv2D.cpp | 105 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 …]
|
D | Neg.cpp | 44 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()
|