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
203 dim1 = getSizeOfDimension(in1, numberOfDims1 - i);
207 dim2 = getSizeOfDimension(in2, numberOfDims2 - i);
254 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0));
255 NN_OPS_CHECK(getSizeOfDimension(filter, 3) == getSizeOfDimension(input, 3));
257 uint32_t channels_out = getSizeOfDimension(filter, 0);
258 uint32_t width = getSizeOfDimension(input, 2);
259 uint32_t height = getSizeOfDimension(inpu
[all...]

Completed in 53 milliseconds