Searched refs:DelegateKernel (Results 1 – 4 of 4) sorted by relevance
27 class DelegateKernel : public SimpleDelegateKernelInterface {29 DelegateKernel();30 ~DelegateKernel() override;
375 DelegateKernel::DelegateKernel() : op_data_(new OpData) {} in DelegateKernel() function in tflite::flex::DelegateKernel376 DelegateKernel::~DelegateKernel() {} in ~DelegateKernel()378 TfLiteStatus DelegateKernel::Init(TfLiteContext* context, in Init()443 TfLiteStatus DelegateKernel::Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()515 TfLiteStatus DelegateKernel::ValidateOutputTensorShapeConsistency( in ValidateOutputTensorShapeConsistency()594 TfLiteStatus DelegateKernel::Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
95 new tflite::flex::DelegateKernel()); in CreateDelegateKernelInterface()
110 friend class DelegateKernel;114 class DelegateKernel { class116 explicit DelegateKernel(Delegate* delegate) : delegate_(delegate) { in DelegateKernel() function in tflite::gpu::__anonfe059bf90111::DelegateKernel119 ~DelegateKernel() { --delegate_->num_delegate_kernels_; } in ~DelegateKernel()358 inline DelegateKernel* GetDelegateKernel(TfLiteNode* node) { in GetDelegateKernel()359 return reinterpret_cast<DelegateKernel*>(node->user_data); in GetDelegateKernel()376 absl::make_unique<DelegateKernel>(gpu_delegate); in DelegatePrepare()387 delete reinterpret_cast<DelegateKernel*>(buffer); in DelegatePrepare()