Home
last modified time | relevance | path

Searched refs:getKernelName (Results 1 – 7 of 7) sorted by relevance

/external/angle/src/libANGLE/renderer/vulkan/
DCLKernelVk.h97 const std::string &getKernelName() { return mName; } in getKernelName() function
DCLCommandQueueVk.cpp1237 devProgramData->getCompiledWorkgroupSize(kernelImpl.getKernelName()); in enqueueNDRangeKernel()
1859 ->getDeviceProgramData(kernelVk.getKernelName().c_str()) in processKernelResources()
1863 mPrintfInfos = kernelVk.getProgram()->getPrintfDescriptors(kernelVk.getKernelName()); in processKernelResources()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc185 auto kernel_name = launch_op.getKernelName().getValue(); in matchAndRewrite()
/external/pytorch/torch/csrc/jit/tensorexpr/
Dkernel.h181 const std::string& getKernelName() const { in getKernelName() function
/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp1362 RECORD_FUNCTION(kernel->getKernelName(), std::vector<c10::IValue>()); in createTensorExprOp()
1423 RECORD_FUNCTION(kernel->getKernelName(), std::vector<c10::IValue>()); in createTensorExprOp()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dlmhlo_to_jitrt.cc379 call->setAttr(b.getStringAttr("kernel"), op.getKernelName()); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1596 launch_func.getKernelName().str()); in EmitLaunchFunc()