diff options
Diffstat (limited to 'tensorflow/contrib/tensorrt/resources/trt_allocator.h')
-rw-r--r-- | tensorflow/contrib/tensorrt/resources/trt_allocator.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/resources/trt_allocator.h b/tensorflow/contrib/tensorrt/resources/trt_allocator.h index 6f94492083..dc9862b16c 100644 --- a/tensorflow/contrib/tensorrt/resources/trt_allocator.h +++ b/tensorflow/contrib/tensorrt/resources/trt_allocator.h @@ -29,7 +29,7 @@ limitations under the License. namespace tensorflow { namespace tensorrt { // std::align is not supported, so this function mimic its behavior. -void* Align(size_t alignment, size_t size, void*& ptr, size_t& space); +void* Align(uint64_t alignment, uint64_t size, void*& ptr, uint64_t& space); } // namespace tensorrt } // namespace tensorflow |