Searched refs:GPUModule (Results 1 – 2 of 2) sorted by relevance
38 using GPUModule = CUmodule;41 using GPUModule = hipModule_t; variable52 GPUModule LookupOrLoadModule(void* data);56 absl::flat_hash_map<void*, GPUModule> gpu_module_by_data_ptr_
90 GPURuntimeCache::GPUModule GPURuntimeCache::LookupOrLoadModule(void *data) { in LookupOrLoadModule()92 GPUModule &module = gpu_module_by_data_ptr_[data]; in LookupOrLoadModule()137 GPURuntimeCache::GPUModule module = cache->LookupOrLoadModule(module_blob); in _mlir_ciface_tf_launch_kernel()