diff --git a/frameworks/native/neural_network_runtime/ops/batch_to_space_nd_builder.cpp b/frameworks/native/neural_network_runtime/ops/batch_to_space_nd_builder.cpp
index b56ffcbf5526678666cd98bb2f8fdddc02c88df9..39d103b01a8f5ce0d7de343999b52dc80f6192ae 100644
--- a/frameworks/native/neural_network_runtime/ops/batch_to_space_nd_builder.cpp
+++ b/frameworks/native/neural_network_runtime/ops/batch_to_space_nd_builder.cpp
@@ -97,6 +97,9 @@ OH_NN_ReturnCode BatchToSpaceNDBuilder::Build(const std::vector<uint32_t>& param
         return returnCode;
     }
 
+    m_inputsIndex = inputsIndex;
+    m_outputsIndex = outputsIndex;
+
     for (int i : paramsIndex) {
         std::shared_ptr<NNTensor> tensor = allTensors[i];
         switch (tensor->GetType()) {