Tensorflow 额外调用内核'；s计算方法

构建一个简单的内核，如下所示

#包括“tensorflow/core/framework/op.h”
#包括“tensorflow/core/framework/op_kernel.h”
#包括“tensorflow/core/framework/tensor.h”
#包括“tensorflow/core/lib/core/errors.h”
#包括“tensorflow/core/lib/core/status.h”
#包括“tensorflow/core/lib/strings/str_util.h”
#包括“tensorflow/core/lib/strings/strcat.h”
#包括“tensorflow/core/framework/shape_inference.h”
#包括“tensorflow/core/platform/logging.h”
命名空间tensorflow{
使用名称空间std；
寄存器_OP（“测试”）
.输入（“in:int32”）
.输出（“输出：int32”）
//.SetIsStateful（）
.Doc（R“Doc（
维尔德
)文件);；
类TestOp:public操作内核{
公众：
显式TestOp（OpKernelConstruction*context）：OpKernel（context）{
}                                                                                                                                                                                            
void Compute（OpKernelContext*context）重写{
日志（INFO）可能不是这里的情况，但请注意，运行时需要能够执行“虚拟运行”——执行内核以进行分析。您可能需要检查日志设置。