This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/bash | |
export USE_NCCL=0 | |
export USE_DISTRIBUTED=1 | |
export USE_QNNPACK=0 | |
export USE_PYTORCH_QNNPACK=0 | |
# Orin is based on Ampere Achitecture | |
export TORCH_CUDA_ARCH_LIST="8.7" |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import android.content.Context | |
import org.tensorflow.lite.Interpreter | |
import timber.log.Timber | |
import java.io.FileInputStream | |
import java.nio.MappedByteBuffer | |
import java.nio.channels.FileChannel | |
import kotlin.system.measureTimeMillis | |
/* | |
make sure to include the following in the app/build.gradle dependencies: |