Searched refs:valueShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 462 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare() argument
463 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
466 const uint32_t columns = getSizeOfDimension(valueShape, 1); in embeddingLookupPrepare()
469 outputShape->type = valueShape.type; in embeddingLookupPrepare()
471 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
472 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()
474 outputShape->offset = valueShape.offset; in embeddingLookupPrepare()
475 outputShape->scale = valueShape.scale; in embeddingLookupPrepare()
481 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare() argument
484 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 313 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
316 const Shape& valueShape, Shape* outputShape, Shape* hitShape);
|