Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp482 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare() argument
485 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
489 const uint32_t keys = getSizeOfDimension(keyShape, 0); in hashtableLookupPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h315 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape,