HomeSort by relevance Sort by last modified time
    Searched refs:set_device_memory_size (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.cc 121 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
  /external/tensorflow/tensorflow/stream_executor/
device_description.h 278 void set_device_memory_size(int64 value) { function in class:stream_executor::internal::DeviceDescriptionBuilder
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.cc 194 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_gpu_executor.cc     [all...]

Completed in 76 milliseconds