Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A 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 {
H A Dkernel_spec.cc22 KernelLoaderSpec::KernelLoaderSpec(absl::string_view kernelname) in KernelLoaderSpec() function in stream_executor::KernelLoaderSpec
27 : KernelLoaderSpec(kernelname), filename_(std::string(filename)) {} in OnDiskKernelLoaderSpec()
39 : KernelLoaderSpec(kernelname), bytes_(bytes) {} in CudaCubinInMemory()
53 : KernelLoaderSpec(kernel_name), in CudaPtxInMemory()
66 : KernelLoaderSpec(kernel_name), in CudaPtxInMemory()
166 : KernelLoaderSpec(kernelname), text_(text) {} in OpenCLTextInMemory()