Home
last modified time | relevance | path

Searched refs:MultiKernelLoaderSpec (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dkernel_spec.cc172 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextOnDisk( in AddOpenCLTextOnDisk()
179 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLBinaryOnDisk( in AddOpenCLBinaryOnDisk()
186 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextInMemory( in AddOpenCLTextInMemory()
193 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxOnDisk( in AddCudaPtxOnDisk()
200 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCubinInMemory( in AddCudaCubinInMemory()
207 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCubinOnDisk( in AddCudaCubinOnDisk()
214 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxInMemory( in AddCudaPtxInMemory()
222 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCompressedPtxInMemory( in AddCudaCompressedPtxInMemory()
230 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxInMemory( in AddCudaPtxInMemory()
239 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCompressedPtxInMemory( in AddCudaCompressedPtxInMemory()
[all …]
H A Dkernel_spec.h273 class MultiKernelLoaderSpec {
275 explicit MultiKernelLoaderSpec(size_t arity);
331 MultiKernelLoaderSpec *AddOpenCLTextOnDisk(absl::string_view filename,
333 MultiKernelLoaderSpec *AddOpenCLBinaryOnDisk(absl::string_view filename,
335 MultiKernelLoaderSpec *AddOpenCLTextInMemory(absl::string_view ocl_text,
337 MultiKernelLoaderSpec *AddCudaPtxOnDisk(absl::string_view filename,
339 MultiKernelLoaderSpec *AddCudaCubinOnDisk(absl::string_view filename,
341 MultiKernelLoaderSpec *AddCudaCubinInMemory(const char *cubin_bytes,
343 MultiKernelLoaderSpec *AddCudaPtxInMemory(absl::string_view ptx,
345 MultiKernelLoaderSpec *AddCudaCompressedPtxInMemory(
[all …]
H A Dstream_executor_pimpl.h107 port::Status GetKernel(const MultiKernelLoaderSpec& spec, KernelBase* kernel);
815 MultiKernelLoaderSpec loader_spec(kernel_base->kNumberOfParameters); in CreateTypedKernel()
H A Dstream_executor_internal.h174 virtual port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()
H A Dstream_executor_pimpl.cc192 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.h53 port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.h59 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_executor.h109 port::Status GetKernel(const MultiKernelLoaderSpec& spec,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.cc334 se::MultiKernelLoaderSpec loader_spec(num_args); in CreateKernel()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc240 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc243 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()