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

space

Browse Source
This commit is contained in:
VoVAllen 2020-02-21 09:33:20 +00:00
parent 295a8b30d5
commit cae654ea99
3 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/eager/dlpack.cc
View File

@ -291,4 +291,4 @@ TFE_TensorHandle* TFE_HandleFromDLPack(void* dlm, TF_Status* status) {
return handle; return handle;
}; };
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/python/dlpack/BUILD
View File

@ -19,4 +19,4 @@ cuda_py_test(
"@absl_py//absl/testing:absltest", "@absl_py//absl/testing:absltest",
"@absl_py//absl/testing:parameterized", "@absl_py//absl/testing:parameterized",
] ]
) )

2
tensorflow/python/dlpack/dlpack.py
View File

@ -22,4 +22,4 @@ def to_dlpack(tf_tensor):
@tf_export("dlpack.from_dlpack") @tf_export("dlpack.from_dlpack")
def from_dlpack(dlcapsule): def from_dlpack(dlcapsule):
return pywrap_tfe.TFE_FromDlpackCapsule(dlcapsule) return pywrap_tfe.TFE_FromDlpackCapsule(dlcapsule)