Searched refs:temporary_memory_manager (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A 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()
|
| H A 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",
|
| H A D | stream.h | 1474 internal::TemporaryMemoryManager *temporary_memory_manager();
1648 inline internal::TemporaryMemoryManager *Stream::temporary_memory_manager() { in temporary_memory_manager() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/ |
| H A 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",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.o.d | 1355 external/tensorflow/tensorflow/compiler/xla/stream_executor/temporary_memory_manager.h \
|