Searched defs:assign_var (Results 1 – 6 of 6) sorted by relevance
62 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite() local100 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
196 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
108 Node* assign_var = nullptr; in TEST() local175 Node* assign_var = nullptr; in CreateSplitResourceGraph() local631 Node* assign_var = nullptr; in TEST() local750 Node* assign_var = nullptr; in TEST() local831 Node* assign_var = nullptr; in CreateConcatResourceGraph() local1285 Node* assign_var = nullptr; in CreateRoundtripResourceGraph() local