Home
last modified time | relevance | path

Searched refs:corepb (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/go/
Dsignature_test.go25 corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto" packageName
29 got := signatureDefFromProto(&corepb.SignatureDef{
30 Inputs: map[string]*corepb.TensorInfo{
31 "input_1": &corepb.TensorInfo{
32 Encoding: &corepb.TensorInfo_Name{
44 "input_2": &corepb.TensorInfo{
45 Encoding: &corepb.TensorInfo_Name{
58 Outputs: map[string]*corepb.TensorInfo{
59 "output_1": &corepb.TensorInfo{
60 Encoding: &corepb.TensorInfo_Name{
[all …]
Dsignature.go19 import corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_prot… packageName
93 func signatureDefFromProto(pb *corepb.SignatureDef) Signature {
109 func tensorInfoFromProto(pb *corepb.TensorInfo) TensorInfo {
Dsaved_model.go25 corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto" packageName
79 metaGraphDef := new(corepb.MetaGraphDef)
94 func generateSignatures(pb map[string]*corepb.SignatureDef) map[string]Signature {