pytorch Architecture Docs
Home / Class/ PyTorchAndroidJni Class — pytorch Architecture

PyTorchAndroidJni Class — pytorch Architecture

Architecture documentation for the PyTorchAndroidJni class in pytorch_jni_common.cpp from the pytorch codebase.

Class cpp NativeJNI

Entity Profile

Relationship Graph

Source Code

android/pytorch_android/src/main/cpp/pytorch_jni_common.cpp lines 670–684

class PyTorchAndroidJni : public facebook::jni::JavaClass<PyTorchAndroidJni> {
 public:
  constexpr static auto kJavaDescriptor = "Lorg/pytorch/PyTorchAndroid;";

  static void registerNatives() {
    javaClassStatic()->registerNatives({
        makeNativeMethod(
            "nativeSetNumThreads", PyTorchAndroidJni::setNumThreads),
    });
  }

  static void setNumThreads(facebook::jni::alias_ref<jclass>, jint numThreads) {
    caffe2::pthreadpool()->set_thread_count(numThreads);
  }
};

Domain

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free