c
天地传奇3
这个作者很懒,什么都没留下…
展开
-
OpenBlas API整理
S:单精度,D:双精度C:单精度复数,Z:双精度复数lda,ldb,ldc:每行的间隔incx,incy:x,y的增量, 一般设置为1#ifndef OPENBLAS_CONST# define OPENBLAS_CONST const#endif#define CBLAS_INDEX size_ttypedef enum CBLAS_ORDER {CblasR...原创 2019-02-20 18:09:40 · 6140 阅读 · 1 评论 -
tensorflow源码阅读-op注册
一、OP注册//tensorflow/core/platform/maxros.h// A macro to disallow the copy constructor and operator= functions// This is usually placed in the private: declarations for a class.#define TF_DISALLOW_C...原创 2019-03-12 14:46:25 · 1662 阅读 · 1 评论 -
tensorflow添加一个device
tensorflow目前支持CPU,GPU,TPU。如果我们需要新的硬件,我们如何去适配新设备呢?下面我们来添加一个假设备SPU,具体的计算还是使用CPU的计算。(tensorflow-1.11.0)一、为SPU向设备工厂添加绑定//tensorflow/core/common_runtime/device_factory.cc Status DeviceFactory::AddDevice...原创 2019-03-17 16:57:44 · 1076 阅读 · 0 评论 -
tensorflow源码阅读-opkernel注册
//tensorflow/core/framework/kernel_def_builder.h#define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ REGISTER_KERNEL_BUILDER_UNIQ_HELPER(__COUNTER__, kernel_builder, __VA_ARGS__)#define REGISTE...原创 2019-03-13 15:26:20 · 1522 阅读 · 4 评论