Home
last modified time | relevance | path

Searched refs:keyShape (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/
DOperationsUtils.cpp517 const Shape &keyShape, in hashtableLookupPrepare() argument
522 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
526 const uint32_t keys = getSizeOfDimension(keyShape, 0); in hashtableLookupPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h324 const Shape &keyShape,