Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DeviceType (0.19 sec)

  1. tensorflow/c/c_api_experimental.cc

                                           const char* device_type,
                                           TF_Status* status) {
      status->status = tensorflow::FindKernelDef(
          tensorflow::DeviceType(device_type), builder->BuildNodeDef(),
          /* def = */ nullptr, /* kernel_class_name = */ nullptr);
    }
    
    const char* TF_GetNumberAttrForOpListInput(const char* op_name, int input_index,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental.cc

      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      return tensorflow::unwrap(h)->DeviceType(&status->status);
    }
    
    int TFE_TensorHandleDeviceID(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
Back to top