Searched refs:total_byte_size_ (Results 1 – 3 of 3) sorted by relevance
43 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in DnnScratchAllocator()68 total_byte_size_ += byte_size; in AllocateBytes()73 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize()77 int64 total_byte_size_; variable
384 total_byte_size_ += byte_size; in AllocateBytes()390 int64 TotalByteSize() const { return total_byte_size_; } in TotalByteSize()397 int64 total_byte_size_ = 0; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp417 CHECK(total_byte_size_ == 0) in AllocateBytes()428 total_byte_size_ += byte_size; in AllocateBytes()434 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize()437 int64 total_byte_size_ = 0; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput458 if (total_byte_size_ != 0) { in AllocateBytes()468 total_byte_size_ += byte_size; in AllocateBytes()471 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize()[all …]
315 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in CufftScratchAllocator()336 total_byte_size_ += byte_size; in AllocateBytes()341 int64 TotalByteSize() { return total_byte_size_; } in TotalByteSize()345 int64 total_byte_size_; member in tensorflow::__anonbbe6ef150111::CufftScratchAllocator