| /external/tensorflow/tensorflow/stream_executor/ |
| D | scratch_allocator.h | 44 virtual int64 GetMemoryLimitInBytes() = 0;
65 int64 GetMemoryLimitInBytes() override;
|
| D | scratch_allocator.cc | 29 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes() { return -1; } in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 35 int64 FftScratchAllocator::GetMemoryLimitInBytes() { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator
43 if (byte_size > GetMemoryLimitInBytes()) { in AllocateBytes()
48 byte_size, GetMemoryLimitInBytes())); in AllocateBytes()
|
| D | fft_thunk.h | 44 int64 GetMemoryLimitInBytes() override;
|
| D | gpu_conv_algorithm_picker.cc | 60 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anon6437d7ed0111::ScratchAllocator
84 if (byte_size > GetMemoryLimitInBytes()) { in AllocateBytes()
89 byte_size, GetMemoryLimitInBytes())); in AllocateBytes()
|
| D | gpu_conv_runner.cc | 52 int64 GetMemoryLimitInBytes() override { return scratch_.size(); } in GetMemoryLimitInBytes() function in xla::gpu::__anon53f8f4230111::ScratchBufAllocator
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 53 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
|
| D | redzone_allocator.cc | 66 if (byte_size > GetMemoryLimitInBytes()) { in AllocateBytes()
71 byte_size, GetMemoryLimitInBytes())); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 56 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
|
| D | cudnn_rnn_ops.cc | 366 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInTemp
411 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInOutput
450 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRNNPersistentSpaceAllocator
|
| D | fft_ops.cc | 367 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
|
| D | fused_batch_norm_op.cc | 671 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
724 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
|
| D | matmul_op_impl.h | 295 int64 GetMemoryLimitInBytes() override { return -1; }
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2711 ? std::max(scratch_allocator->GetMemoryLimitInBytes(), int64{0}) in GetCudnnConvolutionForwardAlgorithm()
2773 ? std::max(scratch_allocator->GetMemoryLimitInBytes(), int64{0}) in GetCudnnConvolutionBackwardDataAlgorithm()
2834 ? std::max(scratch_allocator->GetMemoryLimitInBytes(), int64{0}) in GetCudnnConvolutionBackwardFilterAlgorithm()
|