Home
last modified time | relevance | path

Searched refs:mutable_device_memory (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dtemporary_device_memory.h62 DeviceMemoryBase* mutable_device_memory();
112 DeviceMemory<T>* mutable_device_memory() { in mutable_device_memory() function
115 TemporaryDeviceMemoryBase::mutable_device_memory()); in mutable_device_memory()
H A Dtemporary_device_memory.cc28 DeviceMemoryBase* TemporaryDeviceMemoryBase::mutable_device_memory() { in mutable_device_memory() function in stream_executor::TemporaryDeviceMemoryBase
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.cc1240 a = DeviceMemory<CUDA_T *>(*a_temporary->mutable_device_memory()); in DoBlasInternalImpl()
1241 b = DeviceMemory<CUDA_T *>(*b_temporary->mutable_device_memory()); in DoBlasInternalImpl()
1242 c = DeviceMemory<CUDA_T *>(*c_temporary->mutable_device_memory()); in DoBlasInternalImpl()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc4065 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolForward()
4069 wsp_mem->mutable_device_memory()->opaque()); in DoPoolForward()
4221 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolBackward()
H A Drocm_blas.cc753 DeviceMemory<MAPPED_T>(*(*temp_memory)->mutable_device_memory()); in AllocateStridedBuffer()
/aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_src_trace_processor_tables_tables_python/gen/src/trace_processor/tables/
Dprofiler_tables_py.h8873 return mutable_table()->mutable_device_memory()->Set(row_number_, v); in set_device_memory()
9366 mutable_device_memory()->Append(row.device_memory); in Insert()
9454 TypedColumn<std::optional<int64_t>>* mutable_device_memory() { in mutable_device_memory() function