Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for device_type (0.05 sec)

  1. tensorflow/c/c_api_experimental.cc

    }
    
    void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder,
                                           const char* device_type,
                                           TF_Status* status) {
      status->status = tensorflow::FindKernelDef(
          tensorflow::DeviceType(device_type), builder->BuildNodeDef(),
          /* def = */ nullptr, /* kernel_class_name = */ nullptr);
    }
    
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Mon Aug 18 03:53:25 UTC 2025
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

        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;
        if (string(device_type) == DEVICE_GPU) {
          return device_name;
        }
      }
      // No GPU device found.
      return "";
    }
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Fri Dec 27 12:18:10 UTC 2024
    - 97K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

        return list->response[index].accessor;                                \
      }
    
    TF_DEVICELIST_METHOD(const char*, TF_DeviceListName, name().c_str(), nullptr);
    TF_DEVICELIST_METHOD(const char*, TF_DeviceListType, device_type().c_str(),
                         nullptr);
    TF_DEVICELIST_METHOD(int64_t, TF_DeviceListMemoryBytes, memory_limit(), -1);
    TF_DEVICELIST_METHOD(uint64_t, TF_DeviceListIncarnation, incarnation(), 0);
    
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Tue May 13 06:30:43 UTC 2025
    - 102.3K bytes
    - Viewed (0)
Back to top