Searched refs:HostExecutor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 43 HostExecutor::HostExecutor(const PluginConfig &plugin_config) in HostExecutor() function in stream_executor::host::HostExecutor
46 HostExecutor::~HostExecutor() {} in ~HostExecutor()
48 port::Status HostExecutor::Init(int device_ordinal, in Init()
62 bool HostExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage()
69 DeviceMemoryBase HostExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate()
78 void *HostExecutor::GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes, in GetSubBuffer()
83 void HostExecutor::Deallocate(DeviceMemoryBase *mem) { in Deallocate()
87 port::Status HostExecutor::SynchronousMemZero(DeviceMemoryBase *location, in SynchronousMemZero()
93 port::Status HostExecutor::SynchronousMemSet(DeviceMemoryBase *location, in SynchronousMemSet()
99 bool HostExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
[all …]
|
| D | host_platform.cc | 46 return HostExecutor::CreateDeviceDescription(ordinal); in DescriptionForDevice()
75 this, absl::make_unique<HostExecutor>(config.plugin_config), in GetUncachedExecutor()
|
| D | host_gpu_executor.h | 44 class HostExecutor : public internal::StreamExecutorInterface {
46 explicit HostExecutor(const PluginConfig &plugin_config);
47 ~HostExecutor() override;
|