diff options
Diffstat (limited to 'tensorflow/core/common_runtime/sycl/sycl_allocator.h')
| -rw-r--r-- | tensorflow/core/common_runtime/sycl/sycl_allocator.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tensorflow/core/common_runtime/sycl/sycl_allocator.h b/tensorflow/core/common_runtime/sycl/sycl_allocator.h index c896f7f603..15d9ab41a4 100644 --- a/tensorflow/core/common_runtime/sycl/sycl_allocator.h +++ b/tensorflow/core/common_runtime/sycl/sycl_allocator.h @@ -27,8 +27,8 @@ limitations under the License. namespace tensorflow { class SYCLAllocator : public Allocator { -public: - SYCLAllocator(Eigen::QueueInterface* device) : device_(device) {} + public: + SYCLAllocator(Eigen::QueueInterface *device) : device_(device) {} virtual ~SYCLAllocator() override; string Name() override; void *AllocateRaw(size_t alignment, size_t num_bytes) override; @@ -36,11 +36,12 @@ public: void EnterLameDuckMode(); virtual bool ShouldAllocateEmptyTensors() override final { return true; } -private: + + private: Eigen::QueueInterface *device_; // not owned TF_DISALLOW_COPY_AND_ASSIGN(SYCLAllocator); }; -} // namespace tensorflow +} // namespace tensorflow -#endif // TENSORFLOW_COMMON_RUNTIME_SYCL_SYCL_ALLOCATOR_H_ +#endif // TENSORFLOW_COMMON_RUNTIME_SYCL_SYCL_ALLOCATOR_H_ |