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

  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.cc 114 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
  /external/tensorflow/tensorflow/stream_executor/
device_description.h 305 void set_device_memory_size(uint64 value) { function in class:perftools::gputools::internal::DeviceDescriptionBuilder
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.cc 192 builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc     [all...]

Completed in 178 milliseconds