experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix comments

Browse Source
This commit is contained in:
Zhoulong Jiang 2020-11-17 13:39:12 +00:00
parent 11adf204e8
commit a6e0bd59ad
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/c/BUILD
View File

@ -548,7 +548,7 @@ tf_cuda_library(
":tf_tensor",
"//tensorflow/core:framework",
"//tensorflow/core:framework_lite",
"//tensorflow/c/experimental/stream_executor:stream_executor",
"//tensorflow/c/experimental/stream_executor:stream_executor",
],
}),
)

4
tensorflow/c/kernels.h
View File

@ -67,8 +67,8 @@ typedef struct TF_OpKernelConstruction TF_OpKernelConstruction;
typedef struct TF_OpKernelContext TF_OpKernelContext;
// TF_InitKernel to do op/kernel registration.
// Plugin should either implement TF_InitKernel to register kernels or use
// static registration. This function should register all kernels in a plugin.
// Plugin should implement TF_InitKernel to register kernels. This function
// should register all kernels in a plugin.
void TF_InitKernel();
// Allocates a new kernel builder and returns a pointer to it.