Home
last modified time | relevance | path

Searched defs:ProfilingGraphExecutorImpl (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dprofiling_graph_executor_impl.h14 struct TORCH_API ProfilingGraphExecutorImpl : public GraphExecutorImplBase { struct
27 bool isOptimized() const override { in isOptimized()
49 std::unique_ptr<ProfilingRecord> pr_;
51 profiling_plan_; // plan to run in order to profiling the code
52 std::optional<ExecutionPlan> optimized_plan_;
53 FusionStrategy fusion_strategy_;
56 std::optional<ExecutionPlan> fallback_plan_;
65 std::vector<std::unique_ptr<Function>> fallback_functions_;
66 std::optional<size_t> remaining_bailout_depth_;
68 int32_t time_optimized_plan_created_ = 0;
[all …]
H A Dprofiling_graph_executor_impl.cpp598 ProfilingGraphExecutorImpl::ProfilingGraphExecutorImpl( in ProfilingGraphExecutorImpl() function in torch::jit::ProfilingGraphExecutorImpl