Searched defs:assign_ops (Results 1 – 2 of 2) sorted by relevance
434 auto assign_ops = front.getOps<TF::AssignVariableOp>(); in AppendResourceStoreValueToReturn() local767 auto assign_ops = func_op.front().getOps<TF::AssignVariableOp>(); in LiftArgRetResourcesForFunction() local
562 llvm::SmallVector<mlir::TF::AssignOp, 4> assign_ops; in ConvertReferenceVariableToResourceVariable() local