From 49ba207bdaf810293efeba4d0ce1b1bce56e5804 Mon Sep 17 00:00:00 2001 From: George Karpenkov Date: Mon, 15 Jun 2020 11:40:07 -0700 Subject: [PATCH] [TF/XLA] Remove the wrapping function XlaTensor::RefCountIsOne PiperOrigin-RevId: 316510507 Change-Id: I7bbee7208348a1ae43b11671ecd95fe1e104f280 --- tensorflow/compiler/jit/xla_tensor.cc | 4 ---- tensorflow/compiler/jit/xla_tensor.h | 2 -- 2 files changed, 6 deletions(-) diff --git a/tensorflow/compiler/jit/xla_tensor.cc b/tensorflow/compiler/jit/xla_tensor.cc index e56d2714b0a..6bad1b703b6 100644 --- a/tensorflow/compiler/jit/xla_tensor.cc +++ b/tensorflow/compiler/jit/xla_tensor.cc @@ -27,10 +27,6 @@ namespace tensorflow { return xla_tensor; } -/*static*/ bool XlaTensor::RefCountIsOne(const Tensor& tensor) { - return tensor.RefCountIsOne(); -} - /*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor( const Tensor& tensor) { const XlaTensor* xla_tensor = FromTensor(&tensor); diff --git a/tensorflow/compiler/jit/xla_tensor.h b/tensorflow/compiler/jit/xla_tensor.h index 7f7d97e3b3f..a6de405ec9e 100644 --- a/tensorflow/compiler/jit/xla_tensor.h +++ b/tensorflow/compiler/jit/xla_tensor.h @@ -39,8 +39,6 @@ class XlaTensor { // fails. 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 // which case the returned value is shaped_buffer()->root_buffer(), or a // normal Tensor in which case the returned value is