|
|
|
@@ -102,9 +102,10 @@ extern "C" JNIEXPORT jobject JNICALL Java_com_mindspore_lite_LiteSession_getInpu |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
extern "C" JNIEXPORT jobject JNICALL Java_com_mindspore_lite_LiteSession_getInputsByName(JNIEnv *env, jobject thiz, |
|
|
|
jlong session_ptr, |
|
|
|
jstring node_name) { |
|
|
|
extern "C" JNIEXPORT jobject JNICALL Java_com_mindspore_lite_LiteSession_getInputsByTensorName(JNIEnv *env, |
|
|
|
jobject thiz, |
|
|
|
jlong session_ptr, |
|
|
|
jstring tensor_name) { |
|
|
|
jclass array_list = env->FindClass("java/util/ArrayList"); |
|
|
|
jmethodID array_list_construct = env->GetMethodID(array_list, "<init>", "()V"); |
|
|
|
jobject ret = env->NewObject(array_list, array_list_construct); |
|
|
|
@@ -118,11 +119,9 @@ extern "C" JNIEXPORT jobject JNICALL Java_com_mindspore_lite_LiteSession_getInpu |
|
|
|
return ret; |
|
|
|
} |
|
|
|
auto *lite_session_ptr = static_cast<mindspore::session::LiteSession *>(pointer); |
|
|
|
auto inputs = lite_session_ptr->GetInputsByName(JstringToChar(env, node_name)); |
|
|
|
for (auto input : inputs) { |
|
|
|
jobject tensor_addr = env->NewObject(long_object, long_object_construct, jlong(input)); |
|
|
|
env->CallBooleanMethod(ret, array_list_add, tensor_addr); |
|
|
|
} |
|
|
|
auto input = lite_session_ptr->GetInputsByTensorName(JstringToChar(env, tensor_name)); |
|
|
|
jobject tensor_addr = env->NewObject(long_object, long_object_construct, jlong(input)); |
|
|
|
env->CallBooleanMethod(ret, array_list_add, tensor_addr); |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
|