Searched refs:module_for_func (Results 1 – 6 of 6) sorted by relevance
92 mlir::OwningOpRef<mlir::ModuleOp> module_for_func; in verify() local94 host_module.str(), op->getContext(), &module_for_func); in verify()101 func::FuncOp func = module_for_func->lookupSymbol<func::FuncOp>("host_func"); in verify()
80 OwningOpRef<mlir::ModuleOp> module_for_func = in EncapsulateFuncAndSerialize() local82 SymbolTable symbol_table(module_for_func.get()); in EncapsulateFuncAndSerialize()115 tensorflow::SerializeMlirModule(module_for_func.get()); in EncapsulateFuncAndSerialize()
81 mlir::OwningOpRef<mlir::ModuleOp> module_for_func; in matchAndRewrite() local83 func::FuncOp func = op.GetHostFunc(&module_for_func); in matchAndRewrite()
94 OwningOpRef<mlir::ModuleOp> module_for_func = in EncapsulateFuncAndSerialize() local101 module_for_func.get().getOperation()->setAttr(kVersionsAttr, versions_attr); in EncapsulateFuncAndSerialize()102 SymbolTable symbol_table(module_for_func.get()); in EncapsulateFuncAndSerialize()138 tensorflow::SerializeMlirModule(module_for_func.get()); in EncapsulateFuncAndSerialize()
108 OwningOpRef<mlir::ModuleOp> module_for_func = in EncapsulateFuncAndSerialize() local110 SymbolTable symbol_table(module_for_func.get()); in EncapsulateFuncAndSerialize()114 tensorflow::SerializeMlirModule(module_for_func.get()); in EncapsulateFuncAndSerialize()
971 mlir::OwningOpRef<mlir::ModuleOp> module_for_func; in InferShapeForXlaHostComputeMlir() local972 func::FuncOp func = host_compute_op.GetHostFunc(&module_for_func); in InferShapeForXlaHostComputeMlir()