Searched refs:valueShape (Results 1 – 2 of 2) sorted by relevance
494 bool embeddingLookupPrepare(const Shape &valueShape, in embeddingLookupPrepare() argument497 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()500 const uint32_t rows = getSizeOfDimension(valueShape, 0); in embeddingLookupPrepare()501 const uint32_t columns = getSizeOfDimension(valueShape, 1); in embeddingLookupPrepare()505 outputShape->type = valueShape.type; in embeddingLookupPrepare()507 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()508 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()510 outputShape->offset = valueShape.offset; in embeddingLookupPrepare()511 outputShape->scale = valueShape.scale; in embeddingLookupPrepare()518 const Shape &valueShape, in hashtableLookupPrepare() argument[all …]
319 bool embeddingLookupPrepare(const Shape &valueShape,325 const Shape &valueShape,