Return the index of device the op runs.
tf.raw_ops.DeviceIndex(
device_names, name=None
)
Given a list of device names, this operation returns the index of the device this op runs. The length of the list is returned in two cases: (1) Device does not exist in the given device list. (2) It is in XLA compilation.
| Args | |
|---|---|
device_names | A list of strings. |
name | A name for the operation (optional). |
| Returns | |
|---|---|
A Tensor of type int32. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/DeviceIndex