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

Go: Update generated wrapper functions for TensorFlow ops.

Browse Source 
PiperOrigin-RevId: 313280323
Change-Id: Ic02129a6e8c2684b823122096517c941a3fd2ba6
This commit is contained in:
A. Unique TensorFlower 2020-05-26 15:35:21 -07:00 committed by TensorFlower Gardener
parent 53037dcd66
commit bba3595ebf
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -33265,6 +33265,14 @@ func TPUReplicatedInputIndex(value int64) TPUReplicatedInputAttr {
}
}
// TPUReplicatedInputIsPacked sets the optional is_packed attribute to value.
// If not specified, defaults to false
func TPUReplicatedInputIsPacked(value bool) TPUReplicatedInputAttr {
return func(m optionalAttr) {
m["is_packed"] = value
}
}
// Connects N inputs to an N-way replicated TPU computation.
//
// This operation holds a replicated input to a `tpu.replicate()` computation subgraph.