Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp480 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare() argument
483 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h315 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape,