Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for MakeDeviceSet (0.08 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc

                            std::vector<int64_t>{0, 0, 0, 0, 0, 0, 0, 0},
                            "'device_assignment' has duplicate device coordinate "
                            "(0, 0, 0, 0)")));
    
    std::vector<std::string> MakeDeviceSet(int num_tasks,
                                           int num_devices_per_task) {
      std::vector<std::string> devices{
          "/job:localhost/replica:0/task:0/device:CPU:0"};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 46.8K bytes
    - Viewed (0)
Back to top