Searched refs:shapeFnLib (Results 1 – 2 of 2) sorted by relevance
38 auto remarkShapeFn = [&](shape::FunctionLibraryOp shapeFnLib, Operation *op) { in runOnOperation() argument45 if (auto fn = shapeFnLib.getShapeFunction(op)) { in runOnOperation()
190 auto shapeFnLib = dyn_cast<shape::FunctionLibraryOp>( in verifyOperationAttribute() local192 if (!shapeFnLib) in verifyOperationAttribute()195 for (auto mapping : shapeFnLib.getMapping()) { in verifyOperationAttribute()