Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_TensorHandleBackingDeviceName (0.4 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      // Verify that the mirrors are placed on the component devices.
      std::string first_device = TFE_TensorHandleBackingDeviceName(
          second_components[0].get(), status.get());
      ASSERT_EQ(second_underlying_devices[0], first_device);
      std::string second_device = TFE_TensorHandleBackingDeviceName(
          second_components[1].get(), status.get());
      ASSERT_EQ(second_underlying_devices[1], second_device);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_distributed_test.cc

      const string composite_device_name =
          "/job:localhost/replica:0/task:0/device:COMPOSITE:0";
      EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status),
                composite_device_name);
      EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status),
                composite_device_name);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
      // Register and run a function which returns the sum of 3 variables.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      return tensorflow::unwrap(h)->DeviceName(&status->status);
    }
    
    const char* TFE_TensorHandleBackingDeviceName(TFE_TensorHandle* h,
                                                  TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top