Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 8 of 8 for device_num (0.07 seconds)

  1. tensorflow/c/eager/c_api_test_util.cc

      TFE_OpSetAttrShape(op, "shape", {}, 0, status);
      TFE_OpSetAttrString(op, "container", "localhost", 0);
      TFE_OpSetAttrString(op, "shared_name", "", 0);
      if (!device_name.empty()) {
        TFE_OpSetDevice(op, device_name.c_str(), status);
      }
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_TensorHandle* var_handle = nullptr;
      int num_retvals = 1;
      TFE_Execute(op, &var_handle, &num_retvals, status);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 23.4K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api.cc

    }
    }  // namespace
    
    extern "C" {
    
    bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) {
      return tensorflow::unwrap(ctx)->IsCustomDevice(device_name);
    }
    
    void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device,
                                  const char* device_name, void* device_info,
                                  TF_Status* status) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
  3. tensorflow/c/eager/immediate_execution_tensor_handle.cc

      }
      absl::Status s;
      const char* device_name = DeviceName(&s);
      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()), ", device=\"",
                          device_name, "\")");
    }
    
    absl::Status ImmediateExecutionTensorHandle::SummarizeValue(
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:24:07 GMT 2024
    - 2.1K bytes
    - Click Count (0)
  4. tensorflow/c/eager/custom_device_testutil.cc

    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/core/lib/gtl/cleanup.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace {
    
    struct LoggingDevice {
      std::string device_name;
      std::string underlying_device;
      // Set to true whenever a TensorHandle is copied onto the device
      bool* arrived_flag;
      // Set to true whenever an operation is executed
      bool* executed_flag;
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Dec 20 23:30:21 GMT 2025
    - 8.4K bytes
    - Click Count (0)
  5. tensorflow/c/eager/dlpack.cc

    DLDevice GetDlContext(TFE_TensorHandle* h, TF_Status* status) {
      DLDevice ctx;
      const char* device_name =
          tensorflow::unwrap(h)->BackingDeviceName(&status->status);
      DeviceNameUtils::ParsedName parsed_name;
      tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name);
      std::string device_type = parsed_name.type;
      int device_id = 0;
      if (parsed_name.has_id) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Mar 13 23:41:52 GMT 2025
    - 13K bytes
    - Click Count (0)
  6. tensorflow/c/eager/immediate_execution_context.h

      //===--------------------------------------------------------------------===//
      virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0;
    
      // Returns whether `device_name` is registered as a custom device.
      virtual bool IsCustomDevice(const string& device_name) = 0;
    
      // Register a custom device. It will return error is the device name is
      // already registered.
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 12.3K bytes
    - Click Count (0)
  7. tensorflow/c/eager/c_api_test.cc

        const char* device_name = TFE_OpGetDevice(matmul, status);
        ASSERT_TRUE(strstr(device_name, "GPU:0") != nullptr);
    
        TFE_OpSetDevice(matmul, "CPU:0", status);
        ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
        device_name = TFE_OpGetDevice(matmul, status);
        ASSERT_TRUE(strstr(device_name, "CPU:0") != nullptr);
      }
    
      TFE_DeleteOp(matmul);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 94.6K bytes
    - Click Count (0)
  8. tensorflow/c/c_api_test.cc

      for (int i = 0; i < num_devices; ++i) {
        const char* device_name = TF_DeviceListName(device_list, i, s);
        CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        const char* device_type = TF_DeviceListType(device_list, i, s);
        CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        LOG(INFO) << "Device " << i << " has name " << device_name << ", type "
                  << device_type;
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Jan 07 04:56:09 GMT 2026
    - 97.3K bytes
    - Click Count (0)
Back to Top