Searched refs:print_kernel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | benchmark.py | 21 self.print_kernel = False
122 if args.print_kernel:
140 if args.print_kernel:
|
| /aosp_15_r20/external/mesa3d/src/intel/compiler/ |
| H A D | intel_clc.c | 203 print_kernel(FILE *fp, const char *prefix, in print_kernel() function
464 print_kernel(fp, prefix, &kernel, isa); in output_isa()
467 print_kernel(stdout, prefix, &kernel, isa); in output_isa()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | OperatorEntry.cpp | 607 …auto print_kernel = [&](const char* k_desc, const AnnotatedKernelContainer& jts, bool is_alias_key… in registerKernel() local
626 print_kernel(toString(k), it->second, c10::isAliasDispatchKey(k)); in registerKernel()
|