Searched refs:DelegateKernel (Results 1 – 5 of 5) sorted by relevance
35 class DelegateKernel : public SimpleDelegateKernelInterface {37 DelegateKernel();38 ~DelegateKernel() override;
405 tensorflow::Status DelegateKernel::ExecuteOpKernelRunner( in ExecuteOpKernelRunner()440 DelegateKernel::DelegateKernel() : op_data_(new OpData) {} in DelegateKernel() function in tflite::flex::DelegateKernel441 DelegateKernel::~DelegateKernel() {} in ~DelegateKernel()443 TfLiteStatus DelegateKernel::Init(TfLiteContext* context, in Init()544 TfLiteStatus DelegateKernel::Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()616 TfLiteStatus DelegateKernel::ValidateOutputTensorShapeConsistency( in ValidateOutputTensorShapeConsistency()709 TfLiteStatus DelegateKernel::Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()805 const std::map<int, int>& DelegateKernel::GetTensorReleaseMap() const { in GetTensorReleaseMap()
69 const std::map<int, int>& GetTensorReleaseMap(DelegateKernel* kernel) { in GetTensorReleaseMap()143 DelegateKernel* delegate_kernel = in TEST_F()144 reinterpret_cast<DelegateKernel*>(node_and_reg->first.user_data); in TEST_F()
102 new tflite::flex::DelegateKernel()); in CreateDelegateKernelInterface()
124 friend class DelegateKernel;128 class DelegateKernel { class130 explicit DelegateKernel(Delegate* delegate) : delegate_(delegate) { in DelegateKernel() function in tflite::gpu::__anon28152d6f0111::DelegateKernel133 ~DelegateKernel() { --delegate_->num_delegate_kernels_; } in ~DelegateKernel()499 inline DelegateKernel* GetDelegateKernel(TfLiteNode* node) { in GetDelegateKernel()500 return reinterpret_cast<DelegateKernel*>(node->user_data); in GetDelegateKernel()517 std::make_unique<DelegateKernel>(gpu_delegate); in DelegatePrepare()528 delete reinterpret_cast<DelegateKernel*>(buffer); in DelegatePrepare()