|
|
|
@ -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;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|