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: 354444525
Change-Id: Idfe16a5c950c862ece2f0b58e87c128034d73d89
This commit is contained in:
A. Unique TensorFlower 2021-01-28 18:47:38 -08:00 committed by TensorFlower Gardener
parent 7d33acc040
commit de3d52e832
1 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,21 +38,6 @@ func makeOutputList(op *tf.Operation, start int, output string) ([]tf.Output, in
return list, start + size, nil return list, start + size, nil
} }
// Operator that connects the output of an XLA computation to other consumer graph nodes.
func XlaClusterOutput(scope *Scope, input tf.Output) (outputs tf.Output) {
if scope.Err() != nil {
return
}
opspec := tf.OpSpec{
Type: "XlaClusterOutput",
Input: []tf.Input{
input,
},
}
op := scope.AddOperation(opspec)
return op.Output(0)
}
// An op used by XLA SPMD partitioner to switch from manual partitioning to // An op used by XLA SPMD partitioner to switch from manual partitioning to
// //
// automatic partitioning. It converts the shard-shaped, manually partitioned input // automatic partitioning. It converts the shard-shaped, manually partitioned input