Searched defs:getSizeOfDimension (Results 1 - 1 of 1) sorted by relevance

/frameworks/ml/nn/common/
H A DOperationsUtils.cpp60 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { function in namespace:android::nn
181 dim1 = getSizeOfDimension(in1, numberOfDims1 - i);
185 dim2 = getSizeOfDimension(in2, numberOfDims2 - i);
232 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0));
233 NN_OPS_CHECK(getSizeOfDimension(filter, 3) == getSizeOfDimension(input, 3));
235 uint32_t channels_out = getSizeOfDimension(filter, 0);
236 uint32_t width = getSizeOfDimension(input, 2);
237 uint32_t height = getSizeOfDimension(inpu
[all...]

Completed in 17 milliseconds