Searched refs:MultiKernelLoaderSpec (Results 1 - 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| H A D | kernel_spec.cc | 172 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextOnDisk(
179 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLBinaryOnDisk(
186 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddOpenCLTextInMemory(
193 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxOnDisk(
200 MultiKernelLoaderSpec *MultiKernelLoaderSpec
248 MultiKernelLoaderSpec::MultiKernelLoaderSpec(size_t arity) : arity_(arity) {} function in class:perftools::gputools::MultiKernelLoaderSpec
[all...] |
| H A D | kernel_spec.h | 26 // static const MultiKernelLoaderSpec &SaxpySpec() {
28 // (new MultiKernelLoaderSpec{4 /* = arity */})
274 class MultiKernelLoaderSpec { class in namespace:perftools::gputools
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 *AddCudaPtxOnDis
[all...] |
| H A D | stream_executor_internal.h | 163 virtual bool GetKernel(const MultiKernelLoaderSpec &spec,
|
| H A D | 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);
|
| H A D | stream_executor_pimpl.cc | 215 bool StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | kernel_thunk.h | 79 std::unique_ptr<perftools::gputools::MultiKernelLoaderSpec> loader_spec_
|
| H A D | kernel_thunk.cc | 45 loader_spec_.reset(new se::MultiKernelLoaderSpec(args_.size()));
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.h | 62 bool GetKernel(const MultiKernelLoaderSpec &spec,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.h | 63 bool GetKernel(const MultiKernelLoaderSpec &spec,
|
| H A D | cuda_gpu_executor.cc | 210 bool CUDAExecutor::GetKernel(const MultiKernelLoaderSpec &spec,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| H A D | host_gpu_executor.h | 54 bool GetKernel(const MultiKernelLoaderSpec &spec,
|
Completed in 481 milliseconds