MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Unstage

Function Unstage

tensorflow/go/op/wrappers.go:6011–6035  ·  view source on GitHub ↗

Op is similar to a lightweight Dequeue. The basic functionality is similar to dequeue with many fewer capabilities and options. This Op is optimized for performance.

(scope *Scope, dtypes []tf.DataType, optional ...UnstageAttr)

Source from the content-addressed store, hash-verified

6009// The basic functionality is similar to dequeue with many fewer
6010// capabilities and options. This Op is optimized for performance.
6011func Unstage(scope *Scope, dtypes []tf.DataType, optional ...UnstageAttr) (values []tf.Output) {
6012 if scope.Err() != nil {
6013 return
6014 }
6015 attrs := map[string]interface{}{"dtypes": dtypes}
6016 for _, a := range optional {
6017 a(attrs)
6018 }
6019 opspec := tf.OpSpec{
6020 Type: "Unstage",
6021
6022 Attrs: attrs,
6023 }
6024 op := scope.AddOperation(opspec)
6025 if scope.Err() != nil {
6026 return
6027 }
6028 var idx int
6029 var err error
6030 if values, idx, err = makeOutputList(op, idx, "values"); err != nil {
6031 scope.UpdateErr("Unstage", err)
6032 return
6033 }
6034 return values
6035}
6036
6037// StageAttr is an optional argument to Stage.
6038type StageAttr func(optionalAttr)

Callers

nothing calls this directly

Calls 5

aClass · 0.85
makeOutputListFunction · 0.85
UpdateErrMethod · 0.80
ErrMethod · 0.45
AddOperationMethod · 0.45

Tested by

no test coverage detected