Searched refs:getSizeOfDimension (Results 1 – 12 of 12) sorted by relevance
/frameworks/ml/nn/common/ |
D | OperationsUtils.cpp | 60 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 …]
|
D | CpuExecutor.cpp | 492 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/ |
D | Conv2D.cpp | 35 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); \
|
D | DepthwiseConv2D.cpp | 27 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); \
|
D | Pooling.cpp | 26 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); \
|
D | Reshape.cpp | 39 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()
|
D | Activation.cpp | 75 uint32_t batch_size = getSizeOfDimension(inputShape, 0); in softmaxFloat32() 166 uint32_t batch_size = getSizeOfDimension(inputShape, 0); in softmaxQuant8()
|
D | FullyConnected.cpp | 42 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32()
|
D | StridedSlice.cpp | 44 int32_t dim = static_cast<int32_t>(getSizeOfDimension(inputShape, idx)); in stridedSliceGeneric()
|
D | SimpleMath.cpp | 286 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanGeneric()
|
/frameworks/ml/nn/common/include/ |
D | CpuOperationUtils.h | 37 dims.sizes[i] = static_cast<int>(getSizeOfDimension(shape, src)); in convertShapeToDims()
|
D | OperationsUtils.h | 68 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
|