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

fix android and windows building

Browse Source
This commit is contained in:
VoVAllen 2020-02-21 09:30:28 +00:00
parent 3f218b1cd8
commit 295a8b30d5
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/c/eager/BUILD
View File

@ -360,6 +360,7 @@ filegroup(
exclude = [ exclude = [
"c_api_experimental.cc", "c_api_experimental.cc",
"*test*", "*test*",
"*dlpack*",
], ],
), ),
visibility = ["//visibility:public"], visibility = ["//visibility:public"],

2
tensorflow/c/eager/dlpack.cc
View File

@ -59,6 +59,8 @@ void DLManagedTensorDeleter(DLManagedTensor* arg) {
TFDLManagedTensorCtx* owner = TFDLManagedTensorCtx* owner =
static_cast<TFDLManagedTensorCtx*>(arg->manager_ctx); static_cast<TFDLManagedTensorCtx*>(arg->manager_ctx);
owner->handle->Unref(); owner->handle->Unref();
delete owner->handle;
delete owner->tensor.dl_tensor.shape;
delete owner; delete owner;
} }

6
tensorflow/c/eager/dlpack.h
View File

@ -24,11 +24,11 @@ namespace tensorflow {
const char* const kDlTensorCapsuleName = "dltensor"; const char* const kDlTensorCapsuleName = "dltensor";
void* TFE_HandleToDLPack(TFE_TensorHandle* h, TF_Status* status); TF_CAPI_EXPORT extern void* TFE_HandleToDLPack(TFE_TensorHandle* h, TF_Status* status);
TFE_TensorHandle* TFE_HandleFromDLPack(void* dlm, TF_Status* status); TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_HandleFromDLPack(void* dlm, TF_Status* status);
void TFE_CallDLManagedTensorDeleter(void* dlm_ptr); TF_CAPI_EXPORT extern void TFE_CallDLManagedTensorDeleter(void* dlm_ptr);
} // namespace tensorflow } // namespace tensorflow