Searched refs:module_blob (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | tf_gpu_runtime_wrappers.cc | 110 extern "C" void _mlir_ciface_tf_launch_kernel(void *ctx, void *module_blob, in _mlir_ciface_tf_launch_kernel() argument
137 GPURuntimeCache::GPUModule module = cache->LookupOrLoadModule(module_blob); in _mlir_ciface_tf_launch_kernel()
|
| H A D | tf_gpu_runtime_wrappers.h | 64 void* ctx, void* module_blob, char* kernel_name, intptr_t gridX,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_kernel_to_llvm_pass.cc | 180 Value module_blob = in matchAndRewrite() local
233 context_arg, module_blob, kernel_name_global, adaptor.gridSizeX(), in matchAndRewrite()
|