Home
last modified time | relevance | path

Searched refs:has_cuda_cubin_in_memory (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h32 bool has_cuda_cubin_in_memory() const { return has_cuda_cubin_in_memory_; } in has_cuda_cubin_in_memory() function
34 CHECK(has_cuda_cubin_in_memory()); in cuda_cubin_in_memory()
Dkernel_spec.h284 bool has_cuda_cubin_in_memory() const { in has_cuda_cubin_in_memory() function
303 CHECK(has_cuda_cubin_in_memory()); in cuda_cubin_in_memory()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc236 } else if (spec.has_cuda_cubin_in_memory()) { in GetKernel()
369 if (spec.has_cuda_cubin_in_memory()) { in LoadModule()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc275 if (spec.has_cuda_cubin_in_memory()) { in GetKernel()
365 if (spec.has_cuda_cubin_in_memory()) { in LoadModule()