提交 23fc134a 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Go: Update generated wrapper functions for TensorFlow ops.

PiperOrigin-RevId: 317755766
Change-Id: I105b220b0760792d9f4fa7c1390c1477cdb1c627
上级 27838410
......@@ -11448,6 +11448,11 @@ func AssertNextDataset(scope *Scope, input_dataset tf.Output, transformations tf
}
// Return the index of device the op runs.
//
// 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.
func DeviceIndex(scope *Scope, device_names []string) (index tf.Output) {
if scope.Err() != nil {
return
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册