Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_GetDeviceName (0.12 sec)

  1. tensorflow/c/kernels.h

    // Returns the Device Name of the device that the context possesses.
    //
    // The returned TF_StringView's underlying string is owned by the OpKernel and
    // has the same lifetime as the OpKernel.
    TF_CAPI_EXPORT TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx);
    
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
    // Returns the rendezvous in the context. Not supported on mobile.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.cc

          device->parsed_name(), tensorflow::DeviceType(device->device_type()));
      if (!id.ok()) return -1;
      return *id;
    #endif  // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
    }
    
    TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx) {
      const auto& device_name =
          reinterpret_cast<tensorflow::OpKernelContext*>(ctx)->device()->name();
      TF_StringView device_name_sv;
      device_name_sv.data = device_name.data();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
Back to top