Home
last modified time | relevance | path

Searched refs:generateParamsArray (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc58 Value generateParamsArray(gpu::LaunchFuncOp launch_op,
96 Value ConvertLaunchFuncOpToTfRuntimeCallPattern::generateParamsArray( in generateParamsArray() function in mlir::kernel_gen::transforms::__anon1a4b4e700111::ConvertLaunchFuncOpToTfRuntimeCallPattern
196 auto kernel_params = generateParamsArray(launch_op, operands, rewriter); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertLaunchFuncToRuntimeCalls.cpp248 Value generateParamsArray(gpu::LaunchFuncOp launchOp,
484 Value ConvertLaunchFuncOpToGpuRuntimeCallPattern::generateParamsArray( in generateParamsArray() function in ConvertLaunchFuncOpToGpuRuntimeCallPattern
620 auto kernelParams = generateParamsArray(launchOp, operands, rewriter); in matchAndRewrite()