Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp626 struct GraphExecutorImpl : public GraphExecutorImplBase { struct
627 GraphExecutorImpl( in GraphExecutorImpl() argument
636 const ExecutionPlan& getPlanFor( in getPlanFor()
643 GraphExecutorState getDebugState() override { in getDebugState()
658 const ExecutionPlan& getOrCompileFallback() { in getOrCompileFallback()
668 const ExecutionPlan& getOrCompile(const Stack& stack) { in getOrCompile()
689 ExecutionPlan compileSpec(const ArgumentSpec& spec) { in compileSpec()
783 ArgumentSpecCreator arg_spec_creator_;
787 ExecutionPlan fallback;
792 std::unordered_map<ArgumentSpec, ExecutionPlan> plan_cache;