- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for lookup_type (0.09 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize.cc
Type GetEmbeddingLookupShape(Value lookup, Value value) { auto lookup_type = mlir::cast<ShapedType>(lookup.getType()); if (!lookup_type.hasStaticShape()) { return nullptr; } auto value_type = mlir::cast<ShapedType>(value.getType()); if (!value_type.hasStaticShape() || value_type.getRank() != 2) { return nullptr; } SmallVector<int64_t> new_shape = {lookup_type.getNumElements(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)