Home
last modified time | relevance | path

Searched refs:getSizeOfDimension (Results 1 – 25 of 37) sorted by relevance

12

/packages/modules/NeuralNetworks/common/operations/
DBidirectionalSequenceRNN.cpp66 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims()
67 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims()
68 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in transposeFirstTwoDims()
217 const uint32_t maxTime = getSizeOfDimension(inputShape, 0); in executeTyped()
218 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped()
219 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in executeTyped()
222 auxInputSize = getSizeOfDimension(auxInputShape, 2); in executeTyped()
224 const uint32_t fwNumUnits = getSizeOfDimension(fwWeightsShape, 0); in executeTyped()
225 const uint32_t bwNumUnits = getSizeOfDimension(bwWeightsShape, 0); in executeTyped()
386 timeMajor ? getSizeOfDimension(input, 1) : getSizeOfDimension(input, 0); in prepare()
[all …]
DUnidirectionalSequenceLSTM.cpp204 const uint32_t maxTime = getSizeOfDimension(inputShape, isTimeMajor(context) ? 0 : 1); in prepare()
205 const uint32_t batchSize = getSizeOfDimension(inputShape, isTimeMajor(context) ? 1 : 0); in prepare()
206 const uint32_t inputSize = getSizeOfDimension(inputShape, inputRank - 1); in prepare()
210 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
211 const uint32_t numCells = getSizeOfDimension(inputToOutputShape, 0); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
216 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()
221 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells); in prepare()
222 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 1), inputSize); in prepare()
227 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numCells); in prepare()
[all …]
DUnidirectionalSequenceRNN.cpp50 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims()
51 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims()
52 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in transposeFirstTwoDims()
96 const uint32_t maxTime = getSizeOfDimension(inputShape, 0); in executeTyped()
97 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped()
98 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in executeTyped()
99 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped()
163 timeMajor ? getSizeOfDimension(input, 1) : getSizeOfDimension(input, 0); in prepare()
165 timeMajor ? getSizeOfDimension(input, 0) : getSizeOfDimension(input, 1); in prepare()
166 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare()
[all …]
DConv2D.cpp109 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
110 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
111 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
112 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
134 uint32_t height = getSizeOfDimension(inputShape, 1); \
135 uint32_t width = getSizeOfDimension(inputShape, 2); \
136 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
137 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
138 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
139 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
[all …]
DQLSTM.cpp183 const uint32_t batchSize = getSizeOfDimension(inputShape, 0); in prepare()
184 const uint32_t inputSize = getSizeOfDimension(inputShape, 1); in prepare()
188 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
189 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare()
193 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
194 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()
199 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numUnits); in prepare()
200 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 1), inputSize); in prepare()
205 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numUnits); in prepare()
206 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 1), inputSize); in prepare()
[all …]
DDepthwiseConv2D.cpp98 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
99 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
100 int32_t filter_width = getSizeOfDimension(filterShape, 2); in initialize()
101 int32_t filter_height = getSizeOfDimension(filterShape, 1); in initialize()
123 uint32_t height = getSizeOfDimension(inputShape, 1); \
124 uint32_t width = getSizeOfDimension(inputShape, 2); \
125 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
126 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
127 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
128 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
[all …]
DRoiAlign.cpp70 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc()
71 uint32_t inHeight = getSizeOfDimension(inputShape, 1); in roiAlignNhwc()
72 uint32_t inWidth = getSizeOfDimension(inputShape, 2); in roiAlignNhwc()
73 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiAlignNhwc()
74 uint32_t outHeight = getSizeOfDimension(outputShape, 1); in roiAlignNhwc()
75 uint32_t outWidth = getSizeOfDimension(outputShape, 2); in roiAlignNhwc()
76 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc()
77 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiAlignNhwc()
197 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignQuantNhwc()
198 uint32_t inHeight = getSizeOfDimension(inputShape, 1); in roiAlignQuantNhwc()
[all …]
DRoiPooling.cpp65 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc()
66 uint32_t inHeight = getSizeOfDimension(inputShape, 1); in roiPoolingNhwc()
67 uint32_t inWidth = getSizeOfDimension(inputShape, 2); in roiPoolingNhwc()
68 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiPoolingNhwc()
69 uint32_t outHeight = getSizeOfDimension(outputShape, 1); in roiPoolingNhwc()
70 uint32_t outWidth = getSizeOfDimension(outputShape, 2); in roiPoolingNhwc()
71 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc()
72 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiPoolingNhwc()
238 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare()
239 uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
[all …]
DTransposeConv2D.cpp76 int32_t filterWidth = getSizeOfDimension(filterShape, 2); in initialize()
77 int32_t filterHeight = getSizeOfDimension(filterShape, 1); in initialize()
79 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4); in initialize()
112 uint32_t numBatches = getSizeOfDimension(inputShape, 0); \
113 uint32_t inputHeight = getSizeOfDimension(inputShape, 1); \
114 uint32_t inputWidth = getSizeOfDimension(inputShape, 2); \
115 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \
116 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
117 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
118 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); \
[all …]
DResizeImageOps.cpp70 const int batchSize = getSizeOfDimension(inputShape, 0); in resizeNearestNeighbor()
71 const int inHeight = getSizeOfDimension(inputShape, 1); in resizeNearestNeighbor()
72 const int inWidth = getSizeOfDimension(inputShape, 2); in resizeNearestNeighbor()
73 const int channels = getSizeOfDimension(inputShape, 3); in resizeNearestNeighbor()
74 const int outHeight = getSizeOfDimension(outputShape, 1); in resizeNearestNeighbor()
75 const int outWidth = getSizeOfDimension(outputShape, 2); in resizeNearestNeighbor()
115 int32_t height = static_cast<int32_t>(getSizeOfDimension(outputShape, 1)); in resizeImageOpNhwc()
116 int32_t width = static_cast<int32_t>(getSizeOfDimension(outputShape, 2)); in resizeImageOpNhwc()
241 uint32_t batches = getSizeOfDimension(input, 0); in prepare()
242 uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
[all …]
DGenerateProposals.cpp68 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength; in bboxTransformFloat32()
69 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32()
244 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare()
245 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare()
246 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare()
249 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
251 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
252 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
253 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoShape, 1), 2); in prepare()
[all …]
DGroupedConv2D.cpp34 uint32_t numBatches = getSizeOfDimension(inputShape, 0); \
35 uint32_t inputHeight = getSizeOfDimension(inputShape, 1); \
36 uint32_t inputWidth = getSizeOfDimension(inputShape, 2); \
37 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \
38 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
39 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
40 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); \
41 uint32_t outputHeight = getSizeOfDimension(outputShape, 1); \
42 uint32_t outputWidth = getSizeOfDimension(outputShape, 2); \
43 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); \
DHeatmapMaxKeypoint.cpp97 uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0); in heatmapMaxKeypointFloat32Nhwc()
98 uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 1); in heatmapMaxKeypointFloat32Nhwc()
99 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, 3); in heatmapMaxKeypointFloat32Nhwc()
100 uint32_t boxInfoLength = getSizeOfDimension(boxesShape, 1); in heatmapMaxKeypointFloat32Nhwc()
268 uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0); in prepare()
269 uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 2); in prepare()
270 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, layout ? 1 : 3); in prepare()
271 uint32_t boxInfoLength = getSizeOfDimension(boxesShape, 1); in prepare()
272 NN_RET_CHECK_EQ(getSizeOfDimension(heatmapShape, layout ? 3 : 1), heatmapSize); in prepare()
274 NN_RET_CHECK_EQ(getSizeOfDimension(boxesShape, 0), numBoxes); in prepare()
DSlice.cpp60 std::vector<uint32_t> beginIndex(getSizeOfDimension(beginShape, 0)); in evalGeneric()
117 NN_RET_CHECK_EQ(getSizeOfDimension(beginShape, 0), n_dims); in prepare()
121 NN_RET_CHECK_EQ(getSizeOfDimension(sizeShape, 0), n_dims); in prepare()
132 sliceSize = getSizeOfDimension(inputShape, i) - sliceBegin; in prepare()
134 NN_RET_CHECK_LE(beginData[i], getSizeOfDimension(inputShape, i)); in prepare()
136 NN_RET_CHECK_LE(sliceBegin + sliceSize, getSizeOfDimension(inputShape, i)); in prepare()
DTranspose.cpp56 int32_t permSize = perm == nullptr ? 2 : static_cast<int32_t>(getSizeOfDimension(permShape, 0)); in transposeGeneric()
117 output.dimensions = {getSizeOfDimension(input, 1), getSizeOfDimension(input, 0)}; in prepare()
128 NN_RET_CHECK_EQ(numInputDims, getSizeOfDimension(permShape, 0)); in prepare()
133 outDims[idx] = getSizeOfDimension(input, permData[idx]); in prepare()
DSqueeze.cpp92 if (getSizeOfDimension(inputShape, idx) == 1) { in prepare()
98 int32_t squeezeDimsSize = static_cast<int32_t>(getSizeOfDimension(squeezeDimsShape, 0)); in prepare()
103 getSizeOfDimension(inputShape, current) == 1); in prepare()
117 outDims[outIdx++] = getSizeOfDimension(inputShape, inIdx); in prepare()
DInstanceNormalization.cpp52 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc()
53 uint32_t height = getSizeOfDimension(inputShape, 1); in instanceNormNhwc()
54 uint32_t width = getSizeOfDimension(inputShape, 2); in instanceNormNhwc()
55 uint32_t depth = getSizeOfDimension(inputShape, 3); in instanceNormNhwc()
DLSTM.cpp433 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
434 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
435 : getSizeOfDimension(input_shape, 0); in LSTMEvalFloat32()
436 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat32()
437 const uint32_t numCells = getSizeOfDimension(input_to_output_weights_shape, 0); in LSTMEvalFloat32()
438 const uint32_t outputSize = getSizeOfDimension(recurrent_to_output_weights_shape, 1); in LSTMEvalFloat32()
553 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
554 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16()
555 : getSizeOfDimension(input_shape, 0); in LSTMEvalFloat16()
556 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat16()
[all …]
DStridedSlice.cpp155 NN_OPS_CHECK(getSizeOfDimension(beginShape, 0) == numInputDims); in prepare()
156 NN_OPS_CHECK(getSizeOfDimension(endShape, 0) == numInputDims); in prepare()
157 NN_OPS_CHECK(getSizeOfDimension(stridesShape, 0) == numInputDims); in prepare()
174 int32_t dim = static_cast<int32_t>(getSizeOfDimension(inputShape, idx)); in prepare()
DPooling.cpp82 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
83 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
369 uint32_t batches = getSizeOfDimension(input, 0); in prepare()
370 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
371 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
372 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
DConcatenation.cpp192 uint32_t sumAxis = getSizeOfDimension(input0, axis); in prepare()
199 sumAxis += getSizeOfDimension(input, axis); in prepare()
201 NN_RET_CHECK_EQ(getSizeOfDimension(input0, d), getSizeOfDimension(input, d)); in prepare()
DFullyConnected.cpp66 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32()
203 uint32_t num_units = getSizeOfDimension(weights, 0); in validateShapes()
204 uint32_t input_size = getSizeOfDimension(weights, 1); in validateShapes()
205 uint32_t bias_len = getSizeOfDimension(bias, 0); in validateShapes()
DL2Normalization.cpp53 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normFloat32Impl()
80 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl()
112 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8SignedImpl()
192 const int32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Signed()
/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp158 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension() function
348 dim1 = getSizeOfDimension(in1, numberOfDims1 - i); in calculateBroadcastedShape()
352 dim2 = getSizeOfDimension(in2, numberOfDims2 - i); in calculateBroadcastedShape()
424 uint32_t batches = getSizeOfDimension(input, 0); in depthToSpacePrepare()
425 uint32_t height = getSizeOfDimension(input, 1); in depthToSpacePrepare()
426 uint32_t width = getSizeOfDimension(input, 2); in depthToSpacePrepare()
427 uint32_t channels = getSizeOfDimension(input, 3); in depthToSpacePrepare()
443 uint32_t batches = getSizeOfDimension(input, 0); in spaceToDepthPrepare()
444 uint32_t height = getSizeOfDimension(input, 1); in spaceToDepthPrepare()
445 uint32_t width = getSizeOfDimension(input, 2); in spaceToDepthPrepare()
[all …]
/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h137 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
329 const int firstDim = getSizeOfDimension(shape, 0); in transposeFirstTwoDimensions()
330 const int secondDim = getSizeOfDimension(shape, 1); in transposeFirstTwoDimensions()
333 blockSize *= getSizeOfDimension(shape, i); in transposeFirstTwoDimensions()

12