Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp495 const Shape &lookupShape, in embeddingLookupPrepare() argument
498 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
503 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in embeddingLookupPrepare()
516 bool hashtableLookupPrepare(const Shape &lookupShape, in hashtableLookupPrepare() argument
521 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
525 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in hashtableLookupPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h320 const Shape &lookupShape,
323 bool hashtableLookupPrepare(const Shape &lookupShape,