HomeSort by relevance Sort by last modified time
    Searched defs:GpuExecutor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/stream_executor/gpu/
gpu_executor.h 44 class GpuExecutor : public internal::StreamExecutorInterface {
48 explicit GpuExecutor(const PluginConfig& plugin_config)
60 ~GpuExecutor() override;
341 SE_DISALLOW_COPY_AND_ASSIGN(GpuExecutor);
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 113 GpuContext* ExtractGpuContext(GpuExecutor* cuda_exec) {
118 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) {
119 return static_cast<GpuExecutor*>(stream_exec->implementation());
122 GpuExecutor::~GpuExecutor() {
123 CHECK(kernel_to_gpu_binary_.empty()) << "GpuExecutor has live kernels.";
124 CHECK(gpu_binary_to_module_.empty()) << "GpuExecutor has loaded modules.";
130 port::Status GpuExecutor::Init(int device_ordinal,
153 bool GpuExecutor::FindOnDiskForComputeCapability(
179 bool GpuExecutor::FindOnDiskForISAVersion(absl::string_view filename
    [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_gpu_executor.cc 91 return static_cast<GpuExecutor*>(stream->parent()->implementation())
95 GpuContext* ExtractGpuContext(GpuExecutor* rocm_exec) {
100 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) {
101 return static_cast<GpuExecutor*>(stream_exec->implementation());
104 GpuExecutor::~GpuExecutor() {
114 CHECK(gpu_binary_to_module_.empty()) << "GpuExecutor has loaded modules.";
116 bool GpuExecutor::UnloadModule(ModuleHandle module_handle) {
122 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) {
139 void GpuExecutor::UnloadKernel(const KernelBase* kernel)
    [all...]

Completed in 215 milliseconds