Home
last modified time | relevance | path

Searched refs:module_blob (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_rocm_runtime_wrappers.cc94 void *module_blob, char *kernel_name, in tfKernelGenLaunchKernel() argument
107 hipModule_t module = HipRuntimeCache::get(stream)->loadModule(module_blob); in tfKernelGenLaunchKernel()
Dtf_cuda_runtime_wrappers.cc93 void *module_blob, char *kernel_name, in tfKernelGenLaunchKernel() argument
106 CUmodule module = CudaRuntimeCache::get(stream)->loadModule(module_blob); in tfKernelGenLaunchKernel()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc171 Value module_blob = in matchAndRewrite() local
224 context_arg, module_blob, kernel_name_global, adaptor.gridSizeX(), in matchAndRewrite()