Searched refs:lookupShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 369 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare() argument
371 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
374 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in embeddingLookupPrepare()
387 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare() argument
389 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
393 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in hashtableLookupPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 239 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
241 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape,
|