Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp60 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension() function
203 dim1 = getSizeOfDimension(in1, numberOfDims1 - i); in addMulPrepare()
207 dim2 = getSizeOfDimension(in2, numberOfDims2 - i); in addMulPrepare()
254 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0)); in convPrepare()
255 NN_OPS_CHECK(getSizeOfDimension(filter, 3) == getSizeOfDimension(input, 3)); in convPrepare()
257 uint32_t channels_out = getSizeOfDimension(filter, 0); in convPrepare()
258 uint32_t width = getSizeOfDimension(input, 2); in convPrepare()
259 uint32_t height = getSizeOfDimension(input, 1); in convPrepare()
260 uint32_t filterWidth = getSizeOfDimension(filter, 2); in convPrepare()
261 uint32_t filterHeight = getSizeOfDimension(filter, 1); in convPrepare()
[all …]
DCpuExecutor.cpp492 int32_t input_width = getSizeOfDimension(inputShape, 2); in executeOperation()
493 int32_t input_height = getSizeOfDimension(inputShape, 1); in executeOperation()
494 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
495 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
579 int32_t input_width = getSizeOfDimension(inputShape, 2); in executeOperation()
580 int32_t input_height = getSizeOfDimension(inputShape, 1); in executeOperation()
581 int32_t filter_width = getSizeOfDimension(filterShape, 2); in executeOperation()
582 int32_t filter_height = getSizeOfDimension(filterShape, 1); in executeOperation()
661 int32_t input_width = getSizeOfDimension(inputShape, 2); in executeOperation()
662 int32_t input_height = getSizeOfDimension(inputShape, 1); in executeOperation()
[all …]
/frameworks/ml/nn/common/operations/
DConv2D.cpp35 uint32_t height = getSizeOfDimension(inputShape, 1); \
36 uint32_t width = getSizeOfDimension(inputShape, 2); \
37 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
38 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
39 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
40 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
41 uint32_t inDepth = getSizeOfDimension(inputShape, 3); \
47 im2colDim.sizes[3] = (int)getSizeOfDimension(outputShape, 0); \
48 im2colDim.sizes[2] = (int)getSizeOfDimension(outputShape, 1); \
49 im2colDim.sizes[1] = (int)getSizeOfDimension(outputShape, 2); \
DDepthwiseConv2D.cpp27 uint32_t height = getSizeOfDimension(inputShape, 1); \
28 uint32_t width = getSizeOfDimension(inputShape, 2); \
29 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
30 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
31 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
32 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
DPooling.cpp26 uint32_t height = getSizeOfDimension(inputShape, 1); \
27 uint32_t width = getSizeOfDimension(inputShape, 2); \
28 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
29 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
DReshape.cpp39 int32_t height = (int32_t) getSizeOfDimension(outputShape, 1); in resizeBilinearFloat32()
40 int32_t width = (int32_t) getSizeOfDimension(outputShape, 2); in resizeBilinearFloat32()
205 int32_t permSize = static_cast<int32_t>(getSizeOfDimension(permShape, 0)); in transposeGeneric()
DActivation.cpp75 uint32_t batch_size = getSizeOfDimension(inputShape, 0); in softmaxFloat32()
166 uint32_t batch_size = getSizeOfDimension(inputShape, 0); in softmaxQuant8()
DFullyConnected.cpp42 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32()
DStridedSlice.cpp44 int32_t dim = static_cast<int32_t>(getSizeOfDimension(inputShape, idx)); in stridedSliceGeneric()
DSimpleMath.cpp286 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanGeneric()
/frameworks/ml/nn/common/include/
DCpuOperationUtils.h37 dims.sizes[i] = static_cast<int>(getSizeOfDimension(shape, src)); in convertShapeToDims()
DOperationsUtils.h68 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);