Tensorflow Device 和 DeviceContext

1. 参考TensorFlow Device Contexts, Streams and Context Switching问题部分:In the GPUDevice code, I noticed that one GPUDeviceContext is made per stream.Is the purpose of this so that
摘要由CSDN通过智能技术生成

1. 参考TensorFlow Device Contexts, Streams and Context Switching


问题部分:

In the GPUDevice code, I noticed that one GPUDeviceContext is made per stream.

Is the purpose of this so that every context can control one OpKernelContext and then as the various streams need to be executed, then the contexts can just be switched which handles pushing different data/code onto the GPU and then executing.

Do the various streams get registered as different devices (ie. '/gpu:0' and '/gpu:1')?

Per this, ThreadPoolDevice's don't have contexts, but if I were to add contexts into ThreadPoolDevice, would they fit best as a sort of ThreadContext?

在GPUDevice代码中,我注意到每个流都有一个GPUDeviceContext。

这样做的目的是为了使每个上下文都可以控制一个OpKernelContext,然后当各种流需要执行时,那么上下文只需要切换即可将不同的数据/代码推送到GPU上,然后执行。

各种流是否被注册为不同的设备(即'/ gpu:0'和'/ gpu:1')?

根据这一点,ThreadPoolDevice没有上下文,但是如果要将上下文添加到ThreadPoolDevice中,它们是否适合作为一种ThreadContext?

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值