aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go/genop
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-13 13:31:52 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-13 13:36:06 -0700
commit0689372bf73fee47a914e5ba8dc6253b15dd30c8 (patch)
treeed1195adc3562b2ad1f9708fff72a68b0d08072f /tensorflow/go/genop
parentfc5885026be1fb2feb6a9ac27c6b8dc594e48ea7 (diff)
Internal Google Cleanup
PiperOrigin-RevId: 188921801
Diffstat (limited to 'tensorflow/go/genop')
-rw-r--r--tensorflow/go/genop/internal/api_def_map.go2
-rw-r--r--tensorflow/go/genop/internal/genop.go16
-rw-r--r--tensorflow/go/genop/internal/genop_test.go2
3 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/go/genop/internal/api_def_map.go b/tensorflow/go/genop/internal/api_def_map.go
index 07b689dbba..8600452b47 100644
--- a/tensorflow/go/genop/internal/api_def_map.go
+++ b/tensorflow/go/genop/internal/api_def_map.go
@@ -31,7 +31,7 @@ import (
"unsafe"
"github.com/golang/protobuf/proto"
- pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework"
+ pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework_go_proto"
)
// Encapsulates a collection of API definitions.
diff --git a/tensorflow/go/genop/internal/genop.go b/tensorflow/go/genop/internal/genop.go
index 82f7510f2e..fb81631218 100644
--- a/tensorflow/go/genop/internal/genop.go
+++ b/tensorflow/go/genop/internal/genop.go
@@ -47,7 +47,7 @@ import (
"unsafe"
"github.com/golang/protobuf/proto"
- pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework"
+ pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework_go_proto"
)
// GenerateFunctionsForRegisteredOps writes a Go source code file to w
@@ -359,13 +359,13 @@ type attrWrapper struct {
api *pb.ApiDef_Attr
}
-func (a *attrWrapper) Name() string { return a.api.Name }
-func (a *attrWrapper) RenameTo() string { return a.api.RenameTo }
-func (a *attrWrapper) Description() string { return a.api.Description }
-func (a *attrWrapper) Type() string { return a.op.Type }
-func (a *attrWrapper) IsListAttr() bool { return isListAttr(a.op) }
-func (a *attrWrapper) HasMinimum() bool { return a.op.HasMinimum }
-func (a *attrWrapper) Minimum() int64 { return a.op.Minimum }
+func (a *attrWrapper) Name() string { return a.api.Name }
+func (a *attrWrapper) RenameTo() string { return a.api.RenameTo }
+func (a *attrWrapper) Description() string { return a.api.Description }
+func (a *attrWrapper) Type() string { return a.op.Type }
+func (a *attrWrapper) IsListAttr() bool { return isListAttr(a.op) }
+func (a *attrWrapper) HasMinimum() bool { return a.op.HasMinimum }
+func (a *attrWrapper) Minimum() int64 { return a.op.Minimum }
func (a *attrWrapper) DefaultValue() interface{} { return a.api.DefaultValue }
type argWrapper struct {
diff --git a/tensorflow/go/genop/internal/genop_test.go b/tensorflow/go/genop/internal/genop_test.go
index b3a23dff10..d20d22e0c1 100644
--- a/tensorflow/go/genop/internal/genop_test.go
+++ b/tensorflow/go/genop/internal/genop_test.go
@@ -22,7 +22,7 @@ import (
"testing"
"github.com/golang/protobuf/proto"
- pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework"
+ pb "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal/proto/tensorflow/core/framework_go_proto"
)
// Creates an ApiDef based on opdef and applies overrides