- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for add_virtual_devices (0.29 sec)
-
tensorflow/c/eager/c_api_test_util.cc
num_virtual_gpus > 0) { tensorflow::GPUOptions* gpu_options = server_def.mutable_default_session_config()->mutable_gpu_options(); auto virtual_devices = gpu_options->mutable_experimental()->add_virtual_devices(); for (int i = 0; i < num_virtual_gpus; ++i) { virtual_devices->add_memory_limit_mb(200); } } return server_def; } TFE_TensorHandle* CreateVarHandle(TFE_Context* ctx,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)