Home
last modified time | relevance | path

Searched refs:ProfilerDisableOptions (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dprofiler_legacy.h314 struct TORCH_API ProfilerDisableOptions { struct
315 ProfilerDisableOptions() = default;
316 ProfilerDisableOptions(bool shouldCleanupTLSState, bool shouldConsolidate) in ProfilerDisableOptions() function
334 std::optional<ProfilerDisableOptions> profilerDisableOptions = argument
377 std::optional<ProfilerDisableOptions> profilerDisableOptions =
398 const std::optional<ProfilerDisableOptions> profilerDisableOptions_;
Dprofiler_legacy.cpp432 std::optional<ProfilerDisableOptions> profilerDisableOptions) { in disableProfilerLegacy()
Dinit.cpp392 py::class_<ProfilerDisableOptions>(m, "_ProfilerDisableOptions") in THPAutograd_initExtension()
397 py::arg("profiler_disable_options") = ProfilerDisableOptions()); in THPAutograd_initExtension()
/external/pytorch/torch/csrc/distributed/rpc/
Drequest_callback_no_python.cpp436 ProfilerDisableOptions requestThreadOptions( in processRunWithProfilingReq()
463 ProfilerDisableOptions opts(false, true); in processRunWithProfilingReq()
/external/pytorch/test/cpp/jit/
Dtest_misc.cpp2632 torch::autograd::profiler::ProfilerDisableOptions(false, true); in TEST()
2659 auto opts = torch::autograd::profiler::ProfilerDisableOptions(true, false); in TEST()
2674 opts = torch::autograd::profiler::ProfilerDisableOptions(true, false); in TEST()