Searched defs:assign_var (Results 1 – 6 of 6) sorted by relevance
53 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite() local91 TF::AssignVariableOp assign_var = in SinkResourceWritesIntoParallelExecute() local
89 auto assign_var = in PartitionResourceReadsWrites() local
53 auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); in GetResourceWriteResult() local
197 if (auto assign_var = llvm::dyn_cast<TF::AssignVariableOp>(user)) in GetXlaShardingFromResult() local
201 auto assign_var = NodeWith(Op("AssignVariableOp"), Inputs(_, Out(merge))); in TEST_F() local233 auto assign_var = in TEST_F() local
92 else if (auto assign_var = llvm::dyn_cast_or_null<TF::AssignVariableOp>(op)) in GetGlobalTensorOp() local