HomeSort by relevance Sort by last modified time
    Searched refs:MultiKernelLoaderSpec (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
kernel_spec.cc 172 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextOnDisk(
179 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLBinaryOnDisk(
186 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextInMemory(
193 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxOnDisk(
200 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCubinInMemory
    [all...]
kernel_spec.h 26 // static const MultiKernelLoaderSpec &SaxpySpec() {
28 // (new MultiKernelLoaderSpec{4 /* = arity */})
274 class MultiKernelLoaderSpec {
276 explicit MultiKernelLoaderSpec(size_t arity);
325 // MultiKernelLoaderSpec. Each of these should be used at most once for a
326 // single MultiKernelLoaderSpec object. See file comment for example usage.
332 MultiKernelLoaderSpec *AddOpenCLTextOnDisk(port::StringPiece filename,
334 MultiKernelLoaderSpec *AddOpenCLBinaryOnDisk(port::StringPiece filename,
336 MultiKernelLoaderSpec *AddOpenCLTextInMemory(port::StringPiece ocl_text,
338 MultiKernelLoaderSpec *AddCudaPtxOnDisk(port::StringPiece filename
    [all...]
stream_executor_internal.h 163 virtual bool GetKernel(const MultiKernelLoaderSpec &spec,
stream_executor_pimpl.h 96 // spec: The MultiKernelLoaderSpec is usually generated as a compile-time
99 // MultiKernelLoaderSpec is selected.
105 bool GetKernel(const MultiKernelLoaderSpec &spec, KernelBase *kernel);
    [all...]
stream_executor_pimpl.cc 215 bool StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec,
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
kernel_thunk.h 79 std::unique_ptr<perftools::gputools::MultiKernelLoaderSpec> loader_spec_
kernel_thunk.cc 45 loader_spec_.reset(new se::MultiKernelLoaderSpec(args_.size()));
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.h 62 bool GetKernel(const MultiKernelLoaderSpec &spec,
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.h 63 bool GetKernel(const MultiKernelLoaderSpec &spec,
cuda_gpu_executor.cc 210 bool CUDAExecutor::GetKernel(const MultiKernelLoaderSpec &spec,
    [all...]
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.h 54 bool GetKernel(const MultiKernelLoaderSpec &spec,

Completed in 744 milliseconds