Searched defs:assign_variable (Results 1 – 3 of 3) sorted by relevance
77 auto assign_variable = llvm::dyn_cast<mlir::TF::AssignVariableOp>(user); in IdentifyConnectedAssignVariableOp() local113 mlir::Operation* assign_variable = in runOnOperation() local
27 namespace assign_variable { namespace
118 if (auto assign_variable = dyn_cast<TF::AssignVariableOp>(op)) { in AnalyzeRegion() local