时间戳

首页 > 解决方案 > TensorFlow Lite 的自定义操作是否有最低样本？

tensorflow - TensorFlow Lite 的自定义操作是否有最低样本？

问题描述

显然，我不能只复制粘贴我为 TensorFlow 编写的自定义操作的代码并在 Lite 版本中使用它

我的代码引用了主要代码的内容：

#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/shape_inference.h"
#include "tensorflow/core/framework/op_kernel.h"

using namespace tensorflow;

和

Status MyAddGrad(const Scope& scope, const Operation& op,
                 const std::vector<Output>& grad_inputs,
                 std::Vector<Output>* grad_outputs){

我在任何地方都找不到 TensorFlow Lite 中自定义操作的完整（甚至是最低限度）示例。只有我应该修改哪些文件才能将 Lite 编译为“.aar”

标签： tensorflowtensorflow-lite

解决方案

自定义运算符指南页面显示了如何Sin在 TFLite 中实现自定义操作的规范示例。

您必须为您的操作定义Prepare和Eval函数，如此处所示，然后通过AddCustom调用让 OpResolver 知道。

推荐阅读