Searched refs:temporary_memory_manager (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | temporary_device_memory.cc | 23 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in ~TemporaryDeviceMemoryBase()
42 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in Finalize()
48 return parent_->temporary_memory_manager()->IsFinalized( in IsFinalized()
53 return parent_->temporary_memory_manager()->HasAllocated( in IsAllocated()
|
| D | BUILD | 49 "temporary_memory_manager.h",
229 "temporary_memory_manager.h",
364 ":temporary_memory_manager",
391 "temporary_memory_manager.h",
507 "temporary_memory_manager.h",
589 name = "temporary_memory_manager",
590 srcs = ["temporary_memory_manager.cc"],
591 hdrs = ["temporary_memory_manager.h"],
611 "temporary_memory_manager.h",
653 "temporary_memory_manager.h",
|
| D | stream.h | 1474 internal::TemporaryMemoryManager *temporary_memory_manager();
1648 inline internal::TemporaryMemoryManager *Stream::temporary_memory_manager() { in temporary_memory_manager() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 154 name = "temporary_memory_manager",
155 hdrs = ["temporary_memory_manager.h"],
157 "//tensorflow/compiler/xla/stream_executor:temporary_memory_manager",
330 "temporary_memory_manager.h",
369 "temporary_memory_manager.h",
|