Home
last modified time | relevance | path

Searched refs:KernelLoaderSpec (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel_spec.h71 class KernelLoaderSpec {
73 virtual ~KernelLoaderSpec() {} in ~KernelLoaderSpec()
79 explicit KernelLoaderSpec(absl::string_view kernelname);
86 SE_DISALLOW_COPY_AND_ASSIGN(KernelLoaderSpec);
92 class OnDiskKernelLoaderSpec : public KernelLoaderSpec {
142 class CudaPtxInMemory : public KernelLoaderSpec {
243 class OpenCLTextInMemory : public KernelLoaderSpec {
259 class CudaCubinInMemory : public KernelLoaderSpec {
Dkernel_spec.cc21 KernelLoaderSpec::KernelLoaderSpec(absl::string_view kernelname) in KernelLoaderSpec() function in stream_executor::KernelLoaderSpec
26 : KernelLoaderSpec(kernelname), filename_(std::string(filename)) {} in OnDiskKernelLoaderSpec()
38 : KernelLoaderSpec(kernelname), bytes_(bytes) {} in CudaCubinInMemory()
52 : KernelLoaderSpec(kernel_name), in CudaPtxInMemory()
65 : KernelLoaderSpec(kernel_name), in CudaPtxInMemory()
165 : KernelLoaderSpec(kernelname), text_(text) {} in OpenCLTextInMemory()