Searched refs:init_func_op (Results 1 – 4 of 4) sorted by relevance
87 FuncOp init_func_op = symbol_table.lookup<mlir::FuncOp>( in runOnOperation() local90 if (!init_func_op) { in runOnOperation()95 if (init_func_op.getBlocks().size() != 1) { in runOnOperation()96 init_func_op.emitError("expects exactly one block in the MLIR function"); in runOnOperation()101 init_func_op.getBlocks().front().getOps<VarHandleOp>(); in runOnOperation()
44 FuncOp init_func_op = symbol_table.lookup<mlir::FuncOp>( in runOnOperation() local47 if (!init_func_op) { in runOnOperation()59 builder.create<TFL::CallOnceOp>(func.getLoc(), init_func_op.getName()); in runOnOperation()
85 auto init_func_op = symbol_table.lookup<mlir::FuncOp>( in Verify() local88 if (!init_func_op) in Verify()92 if (!init_func_op.getType().getResults().empty()) in Verify()96 auto exported_names = GetExportedNames(init_func_op); in Verify()463 auto init_func_op = symbol_table.lookup<mlir::FuncOp>( in matchAndRewrite() local470 auto &operations = init_func_op.front().getOperations(); in matchAndRewrite()476 to_remove.push_back(init_func_op); in matchAndRewrite()506 auto init_func_op = symbol_table.lookup<mlir::FuncOp>( in GetSessionInitializerExportedName() local508 auto exported_names = GetExportedNames(init_func_op); in GetSessionInitializerExportedName()
3594 auto init_func_op = sub_symbol_table.lookup<mlir::FuncOp>(target_node_name); in ConvertInitializer() local3595 init_func_op.removeAttr("tf.entry_function"); in ConvertInitializer()3600 DCHECK_EQ(init_func_op.getNumArguments(), assets.size()); in ConvertInitializer()3603 init_func_op.setArgAttr(iter.index(), "tf_saved_model.bound_input", in ConvertInitializer()3609 init_func_op->setAttr( in ConvertInitializer()