Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_kernel_to_llvm_pass.cc109 auto num_kernel_operands = launch_op.getNumKernelOperands(); in generateParamsArray() local
111 loc, launch_op.getOperands().take_back(num_kernel_operands), in generateParamsArray()
112 adaptor.operands().take_back(num_kernel_operands), builder); in generateParamsArray()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc1629 unsigned num_kernel_operands = launch_func.getNumKernelOperands(); in EmitLaunchFunc() local
1630 slices.reserve(num_kernel_operands); in EmitLaunchFunc()
1635 written_operands.take_back(num_kernel_operands))) { in EmitLaunchFunc()