Home
last modified time | relevance | path

Searched defs:launch_func (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.cc40 bool actually_do_autotune, const LaunchFunc& launch_func, in AutotuneConvImpl()
143 auto launch_func = in AutotuneFusedConv() local
322 auto launch_func = in AutotuneUnfusedConv() local
H A Dmatmul_op_fused.cc187 const LaunchFunc& launch_func) { in AutotuneMatmul()
302 auto launch_func = [&](BlasScratchAllocator& scratch_allocator, in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc1590 auto launch_func = mlir::cast<mlir::gpu::LaunchFuncOp>(op); in EmitLaunchFunc() local