diff --git a/tf_adapter_2.x/npu_device/core/op_executors/hooks/make_iterator_op.cpp b/tf_adapter_2.x/npu_device/core/op_executors/hooks/make_iterator_op.cpp index 68a2b28e6bc3a3cb8310c930fd645cc0f55506a3..c780525f8d1094fbc48d8eca0ba5cef9db2e0209 100644 --- a/tf_adapter_2.x/npu_device/core/op_executors/hooks/make_iterator_op.cpp +++ b/tf_adapter_2.x/npu_device/core/op_executors/hooks/make_iterator_op.cpp @@ -124,8 +124,7 @@ static const auto kernel = [](TFE_Context *context, NpuDevice *dev, const tensor dev->CreateIteratorProvider(context, tensor, {dev->device_id}, status); NPU_REQUIRES_TFE_OK(status); - dev->RunGeGraphPin2CpuAnonymous(context, "dp_init_" + handle.name(), dp_init_graph, num_inputs, inputs, 0, - nullptr, status); + dev->RunGeGraphPin2CpuAnonymous(context, "dp_init_" + handle.name(), dp_init_graph, 0, nullptr, 0, nullptr, status); NPU_REQUIRES_TFE_OK(status); } }