Home
last modified time | relevance | path

Searched refs:set_device_memory_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc121 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.h278 void set_device_memory_size(int64 value) { in set_device_memory_size() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc194 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc923 builder.set_device_memory_size(device_memory_size); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1086 builder.set_device_memory_size(device_memory_size); in PopulateDeviceDescription()