Home
last modified time | relevance | path

Searched refs:removeProfilingNodes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dprofiling_record.cpp299 void ProfilingRecord::removeProfilingNodes(Block* b) { in removeProfilingNodes() function in torch::jit::ProfilingRecord
306 removeProfilingNodes(ib); in removeProfilingNodes()
H A Dprofiling_record.h180 TORCH_API static void removeProfilingNodes(Block* b);
H A Dprofiling_graph_executor_impl.cpp511 ProfilingRecord::removeProfilingNodes(gradient.f->block()); in runProfilingOptimizations()
527 ProfilingRecord::removeProfilingNodes(copy->block()); in runProfilingOptimizations()
H A Djit_trace.cpp305 ProfilingRecord::removeProfilingNodes(pr->profiled_graph_->block()); in TraceGraph()
H A Dgraph_executor.cpp1059 ProfilingRecord::removeProfilingNodes(graph->block()); in replaceBlockWithFallbackGraph()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dinsert_guards.cpp13 ProfilingRecord::removeProfilingNodes(graph_->block()); in run()