Searched refs:tail_assign_variable_ops (Results 1 – 1 of 1) sorted by relevance
60 llvm::ArrayRef<TF::AssignVariableOp> tail_assign_variable_ops) { in MoveTailWritesAfterReplicate() argument69 for (auto assign : tail_assign_variable_ops) { in MoveTailWritesAfterReplicate()90 for (auto indexed_assign : llvm::enumerate(tail_assign_variable_ops)) { in MoveTailWritesAfterReplicate()105 SmallVector<TF::AssignVariableOp, 16> tail_assign_variable_ops; in GetTailWritesToReplicateInvariantResourceVars() local118 tail_assign_variable_ops.push_back(assign); in GetTailWritesToReplicateInvariantResourceVars()124 return std::move(tail_assign_variable_ops); in GetTailWritesToReplicateInvariantResourceVars()