experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: Update generated wrapper functions for TensorFlow ops. 

PiperOrigin-RevId: 317755766
Change-Id: I105b220b0760792d9f4fa7c1390c1477cdb1c627
This commit is contained in:
A. Unique TensorFlower 2020-06-22 16:12:37 -07:00 committed by TensorFlower Gardener
parent 27838410a8
commit 23fc134a17
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/go/op/wrappers.go
View File

@ -11448,6 +11448,11 @@ func AssertNextDataset(scope *Scope, input_dataset tf.Output, transformations tf
} }
// Return the index of device the op runs. // 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) { func DeviceIndex(scope *Scope, device_names []string) (index tf.Output) {
if scope.Err() != nil { if scope.Err() != nil {
return return