- 博客(2)
- 资源 (4)
- 收藏
- 关注
原创 tensorRT 中的推理得分函数分析
分析TensorRT中的int8量化代码中的推理结果统计分析代码,输出推理结果中top1和 top5的统计情况int calculateScore(float* batchProb, float* labels, int batchSize, int outputSize, int threshold){ int success = 0; for (int i = 0; i &...
2019-04-30 16:18:16 2925 1
原创 cuda 和 C++ 文件混编Makefile 实现
.PHONY: all cleanCUDA_LIB_DIR := /usr/local/cuda/lib64LIBS += -lcudartLIBS += -lcudaCU_SRCS := $(wildcard *.cu)CU_OBJS := $(CU_SRCS:%.cu=%.o)C_SRCS := $(wildcard *.cpp)C_OBJS := $(C_SRCS:%.c...
2019-04-25 14:16:14 2543
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人