Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for num_devices (0.11 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

      TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get());
      CHECK_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      const int num_devices = TF_DeviceListCount(devices);
      for (int i = 0; i < num_devices; ++i) {
        const string dev_type(TF_DeviceListType(devices, i, status.get()));
        CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

        return signalPassFailure();
    
      // If there is no runtime device information and data format is not explicitly
      // forced, there is nothing to do.
      if (devices.NumDevices() == 0 && force_data_format_.empty()) return;
    
      func.walk([&](LayoutSensitiveInterface layout_sensitive_interface) {
        // Get desired op data format.
        StringRef target_data_format = force_data_format_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
Back to top