Home
last modified time | relevance | path

Searched defs:kernel_tags (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A DKernelDTypeTracer.cpp16 getCalledKernelTags().withLock([&](kernel_tags_type& kernel_tags) { in KernelDTypeTracer()
H A Dtracer.cpp99 const torch::jit::mobile::KernelDTypeTracer::kernel_tags_type& in printDTypesYAML()
H A DTracerRunner.cpp366 [&](KernelDTypeTracer::kernel_tags_type& kernel_tags) { in trace_run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected_test.cc2011 std::vector<std::string> kernel_tags) { in GenerateSparseTestParam()