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

  /external/tensorflow/tensorflow/stream_executor/
kernel_spec.cc 22 KernelLoaderSpec::KernelLoaderSpec(port::StringPiece kernelname)
23 : kernelname_(kernelname.ToString()) {}
26 port::StringPiece kernelname)
27 : KernelLoaderSpec(kernelname), filename_(filename.ToString()) {}
30 port::StringPiece kernelname)
31 : OnDiskKernelLoaderSpec(filename, kernelname) {}
34 port::StringPiece kernelname)
35 : OnDiskKernelLoaderSpec(filename, kernelname) {}
38 port::StringPiece kernelname)
39 : KernelLoaderSpec(kernelname), bytes_(bytes) {
    [all...]
kernel_spec.h 77 const string &kernelname() const { return kernelname_; } function in class:perftools::gputools::KernelLoaderSpec
80 explicit KernelLoaderSpec(port::StringPiece kernelname);
106 port::StringPiece kernelname);
117 CudaPtxOnDisk(port::StringPiece filename, port::StringPiece kernelname);
129 CudaCubinOnDisk(port::StringPiece filename, port::StringPiece kernelname);
157 CudaPtxInMemory(port::StringPiece ptx, port::StringPiece kernelname,
222 OpenCLTextOnDisk(port::StringPiece filename, port::StringPiece kernelname);
234 OpenCLBinaryOnDisk(port::StringPiece filename, port::StringPiece kernelname);
246 OpenCLTextInMemory(port::StringPiece text, port::StringPiece kernelname);
262 CudaCubinInMemory(const char *bytes, port::StringPiece kernelname);
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 214 const string *kernelname; local
219 kernelname = &spec.cuda_cubin_in_memory().kernelname();
242 kernelname = &spec.cuda_ptx_in_memory().kernelname();
253 LOG(FATAL) << "loader spec has no ptx for kernel " << *kernelname;
263 LOG(ERROR) << "failed to load PTX for kernel " << *kernelname;
280 VLOG(2) << "getting function " << *kernelname << " from module " << module;
281 if (!CUDADriver::GetModuleFunction(context_, module, kernelname->c_str(),
292 LOG(WARNING) << "unable to get metadata for kernel " << *kernelname;
    [all...]

Completed in 2630 milliseconds