mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
30e16d6389
Summary: att Test Plan: ci Rollback Plan: Differential Revision: D81731425 Pull Request resolved: https://github.com/pytorch/pytorch/pull/162353 Approved by: https://github.com/yiming0416
23 lines
492 B
C++
23 lines
492 B
C++
#pragma once
|
|
|
|
#include <torch/nativert/executor/ExecutionFrame.h>
|
|
#include <torch/nativert/executor/OpKernel.h>
|
|
|
|
namespace torch::nativert {
|
|
|
|
class ETDelegateExecutor;
|
|
|
|
class ETCallDelegateKernel : public OpKernel {
|
|
public:
|
|
explicit ETCallDelegateKernel(
|
|
const Node* node,
|
|
ETDelegateExecutor& delegateExecutor);
|
|
|
|
void computeInternal(ExecutionFrame& executionFrame) const override final;
|
|
|
|
private:
|
|
ETDelegateExecutor& delegateExecutor_;
|
|
};
|
|
|
|
} // namespace torch::nativert
|