- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DeviceIDs (0.17 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
absl::Span<const DataType> values, TFE_Context* context, TF_Status* status) const; // A parallel tensor with scalar integers numbering component devices. std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context, TF_Status* status) const; // The number of devices operations run on. size_t num_underlying_devices() const { return underlying_devices_.size(); }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
components.emplace_back(t); } return ParallelTensor::FromTensorHandles(*this, std::move(components), status); } std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs( TFE_Context* context, TF_Status* status) const { std::vector<int32_t> ids; ids.reserve(num_underlying_devices()); for (int i = 0; i < num_underlying_devices(); ++i) { ids.push_back(i); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)