Searched refs:profiled_graph_ (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | jit_trace.cpp | 287 for (auto inp : pr->profiled_graph_->inputs()) { in TraceGraph()
304 ProfilingRecord::removeProfileCounter(pr->profiled_graph_->block()); in TraceGraph()
305 ProfilingRecord::removeProfilingNodes(pr->profiled_graph_->block()); in TraceGraph()
306 insertTracingNodes(pr->profiled_graph_->block(), pr.get(), td); in TraceGraph()
307 GRAPH_DUMP("Profiling Graph:", pr->profiled_graph_); in TraceGraph()
308 Code cd(pr->profiled_graph_, ""); in TraceGraph()
311 for (auto out : pr->profiled_graph_->outputs()) { in TraceGraph()
|
| D | profiling_record.h | 183 std::shared_ptr<Graph> profiled_graph_; member
190 return profiled_graph_; in graph()
|
| D | profiling_record.cpp | 83 : profiled_graph_(std::move(g)), profiling_count_(getNumProfiledRuns()) {} in ProfilingRecord()
88 auto pn = new ProfileOp(profiled_graph_.get(), fp); in createProfileNode()
97 auto pn = new ProfileIValueOp(this->profiled_graph_.get(), nullptr); in createProfileIValueNode()
106 auto pn = new ProfileIValueOp(this->profiled_graph_.get(), nullptr); in createProfileIValueNode()
|
| /external/pytorch/test/cpp/jit/ |
| D | test_misc.cpp | 1922 Code cd(pr->profiled_graph_, ""); in TEST()
1925 auto copy = pr->profiled_graph_->copy(); in TEST()
1973 Code cd(pr->profiled_graph_, ""); in TEST()
1976 auto copy = pr->profiled_graph_->copy(); in TEST()
2017 Code cd(pr->profiled_graph_, ""); in TEST()
2027 ->run(*pr->profiled_graph_); in TEST()
2029 auto begin = pr->profiled_graph_->block()->nodes().begin(); in TEST()
2030 auto end = pr->profiled_graph_->block()->nodes().end(); in TEST()
2067 Code cd(pr->profiled_graph_, ""); in TEST()
2073 ->run(*pr->profiled_graph_); in TEST()
[all …]
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | specialize_autogradzero.cpp | 78 insertProfileNodesForSpecializeAutogradZero(pr->profiled_graph_->block(), pr); in InsertProfileNodesForSpecializeAutogradZero()
|