Home
last modified time | relevance | path

Searched refs:kernelname (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel_spec.cc21 KernelLoaderSpec::KernelLoaderSpec(absl::string_view kernelname) in KernelLoaderSpec() argument
22 : kernelname_(std::string(kernelname)) {} in KernelLoaderSpec()
25 absl::string_view kernelname) in OnDiskKernelLoaderSpec() argument
26 : KernelLoaderSpec(kernelname), filename_(std::string(filename)) {} in OnDiskKernelLoaderSpec()
29 absl::string_view kernelname) in CudaPtxOnDisk() argument
30 : OnDiskKernelLoaderSpec(filename, kernelname) {} in CudaPtxOnDisk()
33 absl::string_view kernelname) in CudaCubinOnDisk() argument
34 : OnDiskKernelLoaderSpec(filename, kernelname) {} in CudaCubinOnDisk()
37 absl::string_view kernelname) in CudaCubinInMemory() argument
38 : KernelLoaderSpec(kernelname), bytes_(bytes) {} in CudaCubinInMemory()
[all …]
Dkernel_spec.h76 const std::string &kernelname() const { return kernelname_; } in kernelname() function
79 explicit KernelLoaderSpec(absl::string_view kernelname);
105 absl::string_view kernelname);
116 CudaPtxOnDisk(absl::string_view filename, absl::string_view kernelname);
128 CudaCubinOnDisk(absl::string_view filename, absl::string_view kernelname);
156 CudaPtxInMemory(absl::string_view ptx, absl::string_view kernelname,
221 OpenCLTextOnDisk(absl::string_view filename, absl::string_view kernelname);
233 OpenCLBinaryOnDisk(absl::string_view filename, absl::string_view kernelname);
245 OpenCLTextInMemory(absl::string_view text, absl::string_view kernelname);
261 CudaCubinInMemory(const char *bytes, absl::string_view kernelname);
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc267 const std::string* kernelname; in GetKernel() local
273 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
278 kernelname = &spec.cuda_ptx_in_memory().kernelname(); in GetKernel()
289 LOG(FATAL) << "Loader spec has no ptx for kernel " << *kernelname; in GetKernel()
298 VLOG(2) << "getting function " << *kernelname << " from module " << module; in GetKernel()
299 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
311 kernel->set_name(*kernelname); in GetKernel()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc242 const string* kernelname; in GetKernel() local
254 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
268 VLOG(2) << "getting function " << *kernelname << " from module " << module; in GetKernel()
269 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
281 kernel->set_name(*kernelname); in GetKernel()