diff options
author | 2017-11-15 12:13:13 -0800 | |
---|---|---|
committer | 2017-11-15 12:17:33 -0800 | |
commit | 51e8b01c126d76a161b1957e8e1c6e87e5409910 (patch) | |
tree | 052921d161ed1f843d3820b916f0083c8e0f2fc4 | |
parent | 9adc48d3083d33c3674b02787a2f1beeb66a4583 (diff) |
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 175861269
-rw-r--r-- | tensorflow/go/op/wrappers.go | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 5a6ae4fa5f..b43c978245 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -20553,6 +20553,27 @@ func Sub(scope *Scope, x tf.Output, y tf.Output) (z tf.Output) { return op.Output(0) } +// Writes a `GraphDef` protocol buffer to a `SummaryWriter`. +// +// Arguments: +// writer: Handle of `SummaryWriter`. +// global_step: The step to write the summary for. +// tensor: A scalar string of the serialized tf.GraphDef proto. +// +// Returns the created operation. +func WriteGraphSummary(scope *Scope, writer tf.Output, global_step tf.Output, tensor tf.Output) (o *tf.Operation) { + if scope.Err() != nil { + return + } + opspec := tf.OpSpec{ + Type: "WriteGraphSummary", + Input: []tf.Input{ + writer, global_step, tensor, + }, + } + return scope.AddOperation(opspec) +} + // MaxPool3DGradGradAttr is an optional argument to MaxPool3DGradGrad. type MaxPool3DGradGradAttr func(optionalAttr) |