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

[TF/XLA] Remove the wrapping function XlaTensor::RefCountIsOne

Browse Source 
PiperOrigin-RevId: 316510507
Change-Id: I7bbee7208348a1ae43b11671ecd95fe1e104f280
This commit is contained in:
George Karpenkov 2020-06-15 11:40:07 -07:00 committed by TensorFlower Gardener
parent 555be8943e
commit 49ba207bda
2 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/jit/xla_tensor.cc
View File

@ -27,10 +27,6 @@ namespace tensorflow {
return xla_tensor; return xla_tensor;
} }
/*static*/ bool XlaTensor::RefCountIsOne(const Tensor& tensor) {
return tensor.RefCountIsOne();
}
/*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor( /*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor(
const Tensor& tensor) { const Tensor& tensor) {
const XlaTensor* xla_tensor = FromTensor(&tensor); const XlaTensor* xla_tensor = FromTensor(&tensor);

2
tensorflow/compiler/jit/xla_tensor.h
View File

@ -39,8 +39,6 @@ class XlaTensor {
// fails. // fails.
static XlaTensor* FromTensor(const Tensor* tensor); static XlaTensor* FromTensor(const Tensor* tensor);
static bool RefCountIsOne(const Tensor& tensor);
// Create a DeviceMemoryBase from a Tensor. The Tensor can be an XlaTensor, in // Create a DeviceMemoryBase from a Tensor. The Tensor can be an XlaTensor, in
// which case the returned value is shaped_buffer()->root_buffer(), or a // which case the returned value is shaped_buffer()->root_buffer(), or a
// normal Tensor in which case the returned value is // normal Tensor in which case the returned value is