diff --git a/tensorflow/core/kernels/conv_ops_gpu.h b/tensorflow/core/kernels/conv_ops_gpu.h index 14e7d033eb9..37c3711e4aa 100644 --- a/tensorflow/core/kernels/conv_ops_gpu.h +++ b/tensorflow/core/kernels/conv_ops_gpu.h @@ -64,8 +64,7 @@ class CudnnScratchAllocator : public perftools::gputools::ScratchAllocator { AllocatorAttributes(), allocation_attr)); if (!allocation_status.ok()) { return perftools::gputools::port::StatusOr< - perftools::gputools::DeviceMemory>( - AsDeviceMemory(nullptr, 0)); + perftools::gputools::DeviceMemory>(); } // Hold the reference of the allocated tensors until the end of the // allocator.