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