Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for add_virtual_devices (2.31 sec)

  1. 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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
Back to top