We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 35202fa commit 1ece5abCopy full SHA for 1ece5ab
tensorflow/c/eager/parallel_device/parallel_device_lib.h
@@ -81,6 +81,11 @@ class ParallelDevice {
81
// The number of devices operations run on.
82
size_t num_underlying_devices() const { return underlying_devices_.size(); }
83
84
+ // The devices operations run on.
85
+ const std::vector<std::string>& underlying_devices() const {
86
+ return underlying_devices_;
87
+ }
88
+
89
// Takes a description of a single operation being executed on the
90
// ParallelDevice, and in turn runs one operation per component device with
91
// its corresponding inputs from the input ParallelTensors. Wraps the
0 commit comments