Lines Matching refs:valueShape
369 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare() argument
370 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
373 const uint32_t columns = getSizeOfDimension(valueShape, 1); in embeddingLookupPrepare()
376 outputShape->type = valueShape.type; in embeddingLookupPrepare()
378 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
379 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()
381 outputShape->offset = valueShape.offset; in embeddingLookupPrepare()
382 outputShape->scale = valueShape.scale; in embeddingLookupPrepare()
388 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare() argument
391 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
394 outputShape->type = valueShape.type; in hashtableLookupPrepare()
396 for (uint32_t i = 1; i < getNumberOfDimensions(valueShape); i++) { in hashtableLookupPrepare()
397 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in hashtableLookupPrepare()
399 outputShape->offset = valueShape.offset; in hashtableLookupPrepare()
400 outputShape->scale = valueShape.scale; in hashtableLookupPrepare()