Searched refs:valueShape (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
461 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
464 const uint32_t columns = getSizeOfDimension(valueShape, 1); in embeddingLookupPrepare()
467 outputShape->type = valueShape.type; in embeddingLookupPrepare()
469 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
470 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()
472 outputShape->offset = valueShape.offset; in embeddingLookupPrepare()
473 outputShape->scale = valueShape.scale; in embeddingLookupPrepare()
479 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare() argument
482 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 297 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
300 const Shape& valueShape, Shape* outputShape, Shape* hitShape);
|