Package | Description |
---|---|
org.bytedeco.tensorflow |
Modifier and Type | Method and Description |
---|---|
PerOpGpuDevice |
OpKernelContext.Params.eigen_gpu_device() |
PerOpGpuDevice |
OpKernelContext.Params.eigen_gpu_device() |
PerOpGpuDevice |
DeviceBase.MakeGpuDevice() |
PerOpGpuDevice |
DeviceBase.MakeGpuDevice() |
Modifier and Type | Method and Description |
---|---|
OpKernelContext.Params |
OpKernelContext.Params.eigen_gpu_device(PerOpGpuDevice setter) |
OpKernelContext.Params |
OpKernelContext.Params.eigen_gpu_device(PerOpGpuDevice setter) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
Copyright © 2022. All rights reserved.