Home
last modified time | relevance | path

Searched refs:gpu_mem (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc135 void *gpu_mem = location->opaque(); in MemZero() local
139 [gpu_mem, size]() { memset(gpu_mem, 0, size); }); in MemZero()
145 void *gpu_mem = location->opaque(); in Memset() local
149 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); }); in Memset()
155 void *gpu_mem = location->opaque(); in Memset32() local
159 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); }); in Memset32()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F() local
124 EXPECT_EQ(16777216, gpu_mem.used_memory); in TEST_F()
126 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F() local
151 EXPECT_EQ(20971520, gpu_mem.used_memory); in TEST_F()
153 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
/external/perfetto/protos/perfetto/trace/ftrace/
Dall_protos.gni38 "gpu_mem.proto",
Dftrace_event.proto38 import "protos/perfetto/trace/ftrace/gpu_mem.proto";
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc80 static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsROCmDevicePtr() argument
81 return const_cast<hipDeviceptr_t>(gpu_mem.opaque()); in AsROCmDevicePtr()
85 static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) { in AsROCmDevicePtr() argument
86 return AsROCmDevicePtr(*gpu_mem); in AsROCmDevicePtr()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc104 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsCudaDevicePtr() argument
105 return reinterpret_cast<CUdeviceptr>(gpu_mem.opaque()); in AsCudaDevicePtr()
109 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase* gpu_mem) { in AsCudaDevicePtr() argument
110 return AsCudaDevicePtr(*gpu_mem); in AsCudaDevicePtr()
/external/llvm-project/openmp/libomptarget/plugins/amdgpu/impl/
Dsystem.cpp390 for (auto &gpu_mem : gpu_proc.memories()) { in init_compute_and_memory() local
391 hsa_amd_memory_pool_t pool = gpu_mem.memory(); in init_compute_and_memory()
400 cpu_proc.addMemory(gpu_mem); in init_compute_and_memory()
/external/perfetto/tools/ftrace_proto_gen/
Devent_list337 gpu_mem/gpu_mem_total
/external/mesa3d/docs/drivers/
Dvc4.rst178 In ``config.txt`` edit ``gpu_mem`` to 16, if you do not need video decoding,
179 edit gpu_mem to 64 if you need video decoding.
/external/perfetto/
DAndroid.bp4160 "protos/perfetto/trace/ftrace/gpu_mem.proto",
4211 "external/perfetto/protos/perfetto/trace/ftrace/gpu_mem.gen.cc",
4262 "protos/perfetto/trace/ftrace/gpu_mem.proto",
4313 "external/perfetto/protos/perfetto/trace/ftrace/gpu_mem.gen.h",
4368 "protos/perfetto/trace/ftrace/gpu_mem.proto",
4418 "external/perfetto/protos/perfetto/trace/ftrace/gpu_mem.pb.cc",
4469 "protos/perfetto/trace/ftrace/gpu_mem.proto",
4519 "external/perfetto/protos/perfetto/trace/ftrace/gpu_mem.pb.h",
4574 "protos/perfetto/trace/ftrace/gpu_mem.proto",
4625 "external/perfetto/protos/perfetto/trace/ftrace/gpu_mem.pbzero.cc",
[all …]
DBUILD2659 "protos/perfetto/trace/ftrace/gpu_mem.proto",
/external/perfetto/protos/perfetto/trace/
Dperfetto_trace.proto4254 // Begin of protos/perfetto/trace/ftrace/gpu_mem.proto
4262 // End of protos/perfetto/trace/ftrace/gpu_mem.proto