Home
last modified time | relevance | path

Searched refs:ProfilingGraphExecutorImpl (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dprofiling_graph_executor_impl.cpp406 FusionBehavior ProfilingGraphExecutorImpl::getCurrentBehavior( in getCurrentBehavior()
420 void ProfilingGraphExecutorImpl::runNoGradOptimizations( in runNoGradOptimizations()
481 void ProfilingGraphExecutorImpl::runProfilingOptimizations( in runProfilingOptimizations()
537 void ProfilingGraphExecutorImpl::runProfilingInsensitiveOptimizations( in runProfilingInsensitiveOptimizations()
598 ProfilingGraphExecutorImpl::ProfilingGraphExecutorImpl( in ProfilingGraphExecutorImpl() function in torch::jit::ProfilingGraphExecutorImpl
605 size_t ProfilingGraphExecutorImpl::getInstantiatedBailoutDepth() { in getInstantiatedBailoutDepth()
614 const ExecutionPlan& ProfilingGraphExecutorImpl::getOptimizedPlanFor( in getOptimizedPlanFor()
701 const ExecutionPlan& ProfilingGraphExecutorImpl::getPlanFor( in getPlanFor()
723 GraphExecutorState ProfilingGraphExecutorImpl::getDebugState() { in getDebugState()
773 void ProfilingGraphExecutorImpl::replaceFallbackGraphWithFallbackFunction( in replaceFallbackGraphWithFallbackFunction()
[all …]
H A Dprofiling_graph_executor_impl.h14 struct TORCH_API ProfilingGraphExecutorImpl : public GraphExecutorImplBase { struct
15 ProfilingGraphExecutorImpl(
23 ~ProfilingGraphExecutorImpl() override = default;
H A Dgraph_executor.cpp803 new ProfilingGraphExecutorImpl( in GraphExecutor()
824 new ProfilingGraphExecutorImpl( in GraphExecutor()
850 std::dynamic_pointer_cast<ProfilingGraphExecutorImpl>(pImpl)) { in debugFlushCompilationCache()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_autodiff.cpp320 ProfilingGraphExecutorImpl executor(graph, "linear"); in TEST_F()