Searched defs:assign_var (Results 1 – 7 of 7) sorted by relevance
56 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite() local94 TF::AssignVariableOp assign_var = in SinkResourceWritesIntoParallelExecute() local
90 auto assign_var = in PartitionResourceReadsWrites() local
54 auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); in GetResourceWriteResult() local
251 if (auto assign_var = llvm::dyn_cast<TF::AssignVariableOp>(user)) in GetXlaShardingFromResult() 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
200 auto assign_var = NodeWith(Op("AssignVariableOp"), Inputs(_, Out(merge))); in TEST_F() local232 auto assign_var = in TEST_F() local
2435 def assign_var(): function