diff --git a/tf_adapter/kernels/geop_npu.cc b/tf_adapter/kernels/geop_npu.cc index 8ad0dc141e131bb5123c172c94b63c22f3950e8c..d41494c5a89a838507c9a3bc15a07c3cb5973cab 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_) {