Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc255 } else if (spec.has_cuda_cubin_in_memory()) { in GetKernel()
375 if (spec.has_cuda_cubin_in_memory()) { in LoadModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc251 if (spec.has_cuda_cubin_in_memory()) { in GetKernel()
333 if (spec.has_cuda_cubin_in_memory()) { in LoadModule()