From d4cfc0827ea2387e909d1ed17f550b442bb55136 Mon Sep 17 00:00:00 2001
From: xiaguozheng <xiaguozheng1@hisilicon.com>
Date: Fri, 21 Apr 2023 16:52:05 +0800
Subject: [PATCH] =?UTF-8?q?fixed=20e32f355=20from=20https://gitee.com/xyy?=
 =?UTF-8?q?=5Fyyds/tensorflow/pulls/2233=20=E6=B7=BB=E5=8A=A0iterations=5F?=
 =?UTF-8?q?per=5Floop?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tf_adapter/kernels/geop_npu.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tf_adapter/kernels/geop_npu.cc b/tf_adapter/kernels/geop_npu.cc
index 8ad0dc141..d41494c5a 100644
--- a/tf_adapter/kernels/geop_npu.cc
+++ b/tf_adapter/kernels/geop_npu.cc
@@ -846,6 +846,7 @@ void GeOp::ComputeAsync(OpKernelContext *ctx, DoneCallback done) {
     ge::Graph ge_graph = ge::GraphUtilsEx::CreateGraphFromComputeGraph(compute_graph);
     if (iteration_per_loop_ > 1) {
       ge_graph.SetNeedIteration(this->need_iteration_);
+      graph_options_["iterations_per_loop"] = std::to_string(iteration_per_loop_);
     }
 
     if (is_host_graph_) {
-- 
Gitee