Home
last modified time | relevance | path

Searched defs:GetMemoryLimitInBytes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dscratch_allocator.h68 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function
90 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dredzone_allocator.h52 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h52 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
Dfft_ops.cc378 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon686d0f5a0111::CufftScratchAllocator
Dfused_batch_norm_op.cc681 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
734 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
Dcudnn_rnn_ops.cc369 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon6bcc4dbf0111::CudnnRnnAllocatorInTemp
414 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon6bcc4dbf0111::CudnnRnnAllocatorInOutput
453 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon6bcc4dbf0111::CudnnRNNSpaceAllocator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc67 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anon8ba6008b0111::ScratchAllocator