Searched refs:lookupShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 460 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare() argument
462 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
465 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in embeddingLookupPrepare()
478 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare() argument
480 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
484 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in hashtableLookupPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 297 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
299 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape,
|