Method WithDevice

tensorflow/go/op/scope.go:142–151

WithDevice returns a new Scope which will cause all operations added to the graph to execute on devices that match the provided device specification. For example, WithDevice("/device:GPU:0") will cause operations added to the graph to execute on GPU #0. An empty string removes any device restricti

(device string)

//
// An empty string removes any device restrictions.
func (s *Scope) WithDevice(device string) *Scope {
	return &Scope{
		graph:               s.graph,
		namemap:             s.namemap,
		namespace:           s.namespace,
		controlDependencies: s.controlDependencies,
		device:              device,
		err:                 s.err,
	}
}

// Err returns the error, if any, encountered during the construction
// of the Graph managed by s.