- 博客(2)
- 收藏
- 关注
原创 Tensorflow中的内存分配
Tensorflow系统复杂,支持平台多,各类数据结构也多,所以设计一个统一的内存管理分配接口很重要。本文主要探讨tensorflow中的内存分配的相关机制,会重点研究其中实现的两种分配算法。内存分配是系统中非常重要的一环,大家平常接触最多的就是malloc(new)和free(delete)。也就是分配和施放,在tensorflow中也是这样。Tensorflow提供了一个公共的接口类,Allocator类,该类提供了两个重要的方法:AllocateRaw分配内存,DeallocateRaw施
2021-06-07 20:46:01 1187
原创 Tensorflow中设计模式-反射模式
Tensorflow中设计模式-反射模式作为一个优秀的深度学习框架,tensorflow支持各种各样的设备和op,而且支持用户自定义op。用户在自定义一个op后,这个op如何注册进tensorflow框架,用户在使用这个op时,tensorflow是如何知道这个op的呢?今天可以聊聊tensorflow底层代码中广泛使用的反射设计模式。在开始聊tensorflow之前,我们先了解一下什么是反射模式。在很多时候,我们创建要创建的对象,在代码编译前都已经写好,但是在有的场合下,只有在程序运行的时候,才知
2021-06-07 18:51:39 431
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人