pkg/tensorflow/core/framework/function.pb.go (318 lines of code) (raw):

// Code generated by protoc-gen-go. DO NOT EDIT. // versions: // protoc-gen-go v1.28.1 // protoc v3.21.8 // source: tensorflow/core/framework/function.proto package framework import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" sync "sync" ) const ( // Verify that this generated code is sufficiently up-to-date. _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) // Verify that runtime/protoimpl is sufficiently up-to-date. _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) ) // A library is a set of named functions. type FunctionDefLibrary struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields Function []*FunctionDef `protobuf:"bytes,1,rep,name=function,proto3" json:"function,omitempty"` Gradient []*GradientDef `protobuf:"bytes,2,rep,name=gradient,proto3" json:"gradient,omitempty"` } func (x *FunctionDefLibrary) Reset() { *x = FunctionDefLibrary{} if protoimpl.UnsafeEnabled { mi := &file_tensorflow_core_framework_function_proto_msgTypes[0] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } } func (x *FunctionDefLibrary) String() string { return protoimpl.X.MessageStringOf(x) } func (*FunctionDefLibrary) ProtoMessage() {} func (x *FunctionDefLibrary) ProtoReflect() protoreflect.Message { mi := &file_tensorflow_core_framework_function_proto_msgTypes[0] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { ms.StoreMessageInfo(mi) } return ms } return mi.MessageOf(x) } // Deprecated: Use FunctionDefLibrary.ProtoReflect.Descriptor instead. func (*FunctionDefLibrary) Descriptor() ([]byte, []int) { return file_tensorflow_core_framework_function_proto_rawDescGZIP(), []int{0} } func (x *FunctionDefLibrary) GetFunction() []*FunctionDef { if x != nil { return x.Function } return nil } func (x *FunctionDefLibrary) GetGradient() []*GradientDef { if x != nil { return x.Gradient } return nil } // A function can be instantiated when the runtime can bind every attr // with a value. When a GraphDef has a call to a function, it must // have binding for every attr defined in the signature. // // TODO(zhifengc): // - device spec, etc. type FunctionDef struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields // The definition of the function's name, arguments, return values, // attrs etc. Signature *OpDef `protobuf:"bytes,1,opt,name=signature,proto3" json:"signature,omitempty"` // Attributes specific to this function definition. Attr map[string]*AttrValue `protobuf:"bytes,5,rep,name=attr,proto3" json:"attr,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` // By convention, "op" in node_def is resolved by consulting with a // user-defined library first. If not resolved, "func" is assumed to // be a builtin op. NodeDef []*NodeDef `protobuf:"bytes,3,rep,name=node_def,json=nodeDef,proto3" json:"node_def,omitempty"` // A mapping from the output arg names from `signature` to the // outputs from `node_def` that should be returned by the function. Ret map[string]string `protobuf:"bytes,4,rep,name=ret,proto3" json:"ret,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` } func (x *FunctionDef) Reset() { *x = FunctionDef{} if protoimpl.UnsafeEnabled { mi := &file_tensorflow_core_framework_function_proto_msgTypes[1] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } } func (x *FunctionDef) String() string { return protoimpl.X.MessageStringOf(x) } func (*FunctionDef) ProtoMessage() {} func (x *FunctionDef) ProtoReflect() protoreflect.Message { mi := &file_tensorflow_core_framework_function_proto_msgTypes[1] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { ms.StoreMessageInfo(mi) } return ms } return mi.MessageOf(x) } // Deprecated: Use FunctionDef.ProtoReflect.Descriptor instead. func (*FunctionDef) Descriptor() ([]byte, []int) { return file_tensorflow_core_framework_function_proto_rawDescGZIP(), []int{1} } func (x *FunctionDef) GetSignature() *OpDef { if x != nil { return x.Signature } return nil } func (x *FunctionDef) GetAttr() map[string]*AttrValue { if x != nil { return x.Attr } return nil } func (x *FunctionDef) GetNodeDef() []*NodeDef { if x != nil { return x.NodeDef } return nil } func (x *FunctionDef) GetRet() map[string]string { if x != nil { return x.Ret } return nil } // GradientDef defines the gradient function of a function defined in // a function library. // // A gradient function g (specified by gradient_func) for a function f // (specified by function_name) must follow the following: // // The function 'f' must be a numerical function which takes N inputs // and produces M outputs. Its gradient function 'g', which is a // function taking N + M inputs and produces N outputs. // // I.e. if we have // // (y1, y2, ..., y_M) = f(x1, x2, ..., x_N), // // then, g is // // (dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N, // dL/dy1, dL/dy2, ..., dL/dy_M), // // where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the // loss function). dL/dx_i is the partial derivative of L with respect // to x_i. type GradientDef struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields FunctionName string `protobuf:"bytes,1,opt,name=function_name,json=functionName,proto3" json:"function_name,omitempty"` // The function name. GradientFunc string `protobuf:"bytes,2,opt,name=gradient_func,json=gradientFunc,proto3" json:"gradient_func,omitempty"` // The gradient function's name. } func (x *GradientDef) Reset() { *x = GradientDef{} if protoimpl.UnsafeEnabled { mi := &file_tensorflow_core_framework_function_proto_msgTypes[2] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } } func (x *GradientDef) String() string { return protoimpl.X.MessageStringOf(x) } func (*GradientDef) ProtoMessage() {} func (x *GradientDef) ProtoReflect() protoreflect.Message { mi := &file_tensorflow_core_framework_function_proto_msgTypes[2] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { ms.StoreMessageInfo(mi) } return ms } return mi.MessageOf(x) } // Deprecated: Use GradientDef.ProtoReflect.Descriptor instead. func (*GradientDef) Descriptor() ([]byte, []int) { return file_tensorflow_core_framework_function_proto_rawDescGZIP(), []int{2} } func (x *GradientDef) GetFunctionName() string { if x != nil { return x.FunctionName } return "" } func (x *GradientDef) GetGradientFunc() string { if x != nil { return x.GradientFunc } return "" } var File_tensorflow_core_framework_function_proto protoreflect.FileDescriptor var file_tensorflow_core_framework_function_proto_rawDesc = []byte{ 0x0a, 0x28, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0x2f, 0x66, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x0a, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x1a, 0x2a, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0x2f, 0x61, 0x74, 0x74, 0x72, 0x5f, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x28, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0x2f, 0x6e, 0x6f, 0x64, 0x65, 0x5f, 0x64, 0x65, 0x66, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x26, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0x2f, 0x6f, 0x70, 0x5f, 0x64, 0x65, 0x66, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x7e, 0x0a, 0x12, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x65, 0x66, 0x4c, 0x69, 0x62, 0x72, 0x61, 0x72, 0x79, 0x12, 0x33, 0x0a, 0x08, 0x66, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x18, 0x01, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x17, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x65, 0x66, 0x52, 0x08, 0x66, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x33, 0x0a, 0x08, 0x67, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x18, 0x02, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x17, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x47, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x44, 0x65, 0x66, 0x52, 0x08, 0x67, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x22, 0xe7, 0x02, 0x0a, 0x0b, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x65, 0x66, 0x12, 0x2f, 0x0a, 0x09, 0x73, 0x69, 0x67, 0x6e, 0x61, 0x74, 0x75, 0x72, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x11, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x4f, 0x70, 0x44, 0x65, 0x66, 0x52, 0x09, 0x73, 0x69, 0x67, 0x6e, 0x61, 0x74, 0x75, 0x72, 0x65, 0x12, 0x35, 0x0a, 0x04, 0x61, 0x74, 0x74, 0x72, 0x18, 0x05, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x21, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x65, 0x66, 0x2e, 0x41, 0x74, 0x74, 0x72, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x04, 0x61, 0x74, 0x74, 0x72, 0x12, 0x2e, 0x0a, 0x08, 0x6e, 0x6f, 0x64, 0x65, 0x5f, 0x64, 0x65, 0x66, 0x18, 0x03, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x13, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x4e, 0x6f, 0x64, 0x65, 0x44, 0x65, 0x66, 0x52, 0x07, 0x6e, 0x6f, 0x64, 0x65, 0x44, 0x65, 0x66, 0x12, 0x32, 0x0a, 0x03, 0x72, 0x65, 0x74, 0x18, 0x04, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x20, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x65, 0x66, 0x2e, 0x52, 0x65, 0x74, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x03, 0x72, 0x65, 0x74, 0x1a, 0x4e, 0x0a, 0x09, 0x41, 0x74, 0x74, 0x72, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x12, 0x10, 0x0a, 0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, 0x12, 0x2b, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x15, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x41, 0x74, 0x74, 0x72, 0x56, 0x61, 0x6c, 0x75, 0x65, 0x52, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x1a, 0x36, 0x0a, 0x08, 0x52, 0x65, 0x74, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x12, 0x10, 0x0a, 0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, 0x12, 0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x4a, 0x04, 0x08, 0x02, 0x10, 0x03, 0x22, 0x57, 0x0a, 0x0b, 0x47, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x44, 0x65, 0x66, 0x12, 0x23, 0x0a, 0x0d, 0x66, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x66, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x4e, 0x61, 0x6d, 0x65, 0x12, 0x23, 0x0a, 0x0d, 0x67, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x5f, 0x66, 0x75, 0x6e, 0x63, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x67, 0x72, 0x61, 0x64, 0x69, 0x65, 0x6e, 0x74, 0x46, 0x75, 0x6e, 0x63, 0x42, 0x6e, 0x0a, 0x18, 0x6f, 0x72, 0x67, 0x2e, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2e, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0x42, 0x0e, 0x46, 0x75, 0x6e, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x72, 0x6f, 0x74, 0x6f, 0x73, 0x50, 0x01, 0x5a, 0x3d, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x74, 0x65, 0x6e, 0x73, 0x6f, 0x72, 0x66, 0x6c, 0x6f, 0x77, 0x2f, 0x67, 0x6f, 0x2f, 0x63, 0x6f, 0x72, 0x65, 0x2f, 0x66, 0x72, 0x61, 0x6d, 0x65, 0x77, 0x6f, 0x72, 0x6b, 0xf8, 0x01, 0x01, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } var ( file_tensorflow_core_framework_function_proto_rawDescOnce sync.Once file_tensorflow_core_framework_function_proto_rawDescData = file_tensorflow_core_framework_function_proto_rawDesc ) func file_tensorflow_core_framework_function_proto_rawDescGZIP() []byte { file_tensorflow_core_framework_function_proto_rawDescOnce.Do(func() { file_tensorflow_core_framework_function_proto_rawDescData = protoimpl.X.CompressGZIP(file_tensorflow_core_framework_function_proto_rawDescData) }) return file_tensorflow_core_framework_function_proto_rawDescData } var file_tensorflow_core_framework_function_proto_msgTypes = make([]protoimpl.MessageInfo, 5) var file_tensorflow_core_framework_function_proto_goTypes = []interface{}{ (*FunctionDefLibrary)(nil), // 0: tensorflow.FunctionDefLibrary (*FunctionDef)(nil), // 1: tensorflow.FunctionDef (*GradientDef)(nil), // 2: tensorflow.GradientDef nil, // 3: tensorflow.FunctionDef.AttrEntry nil, // 4: tensorflow.FunctionDef.RetEntry (*OpDef)(nil), // 5: tensorflow.OpDef (*NodeDef)(nil), // 6: tensorflow.NodeDef (*AttrValue)(nil), // 7: tensorflow.AttrValue } var file_tensorflow_core_framework_function_proto_depIdxs = []int32{ 1, // 0: tensorflow.FunctionDefLibrary.function:type_name -> tensorflow.FunctionDef 2, // 1: tensorflow.FunctionDefLibrary.gradient:type_name -> tensorflow.GradientDef 5, // 2: tensorflow.FunctionDef.signature:type_name -> tensorflow.OpDef 3, // 3: tensorflow.FunctionDef.attr:type_name -> tensorflow.FunctionDef.AttrEntry 6, // 4: tensorflow.FunctionDef.node_def:type_name -> tensorflow.NodeDef 4, // 5: tensorflow.FunctionDef.ret:type_name -> tensorflow.FunctionDef.RetEntry 7, // 6: tensorflow.FunctionDef.AttrEntry.value:type_name -> tensorflow.AttrValue 7, // [7:7] is the sub-list for method output_type 7, // [7:7] is the sub-list for method input_type 7, // [7:7] is the sub-list for extension type_name 7, // [7:7] is the sub-list for extension extendee 0, // [0:7] is the sub-list for field type_name } func init() { file_tensorflow_core_framework_function_proto_init() } func file_tensorflow_core_framework_function_proto_init() { if File_tensorflow_core_framework_function_proto != nil { return } file_tensorflow_core_framework_attr_value_proto_init() file_tensorflow_core_framework_node_def_proto_init() file_tensorflow_core_framework_op_def_proto_init() if !protoimpl.UnsafeEnabled { file_tensorflow_core_framework_function_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { switch v := v.(*FunctionDefLibrary); i { case 0: return &v.state case 1: return &v.sizeCache case 2: return &v.unknownFields default: return nil } } file_tensorflow_core_framework_function_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { switch v := v.(*FunctionDef); i { case 0: return &v.state case 1: return &v.sizeCache case 2: return &v.unknownFields default: return nil } } file_tensorflow_core_framework_function_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { switch v := v.(*GradientDef); i { case 0: return &v.state case 1: return &v.sizeCache case 2: return &v.unknownFields default: return nil } } } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_tensorflow_core_framework_function_proto_rawDesc, NumEnums: 0, NumMessages: 5, NumExtensions: 0, NumServices: 0, }, GoTypes: file_tensorflow_core_framework_function_proto_goTypes, DependencyIndexes: file_tensorflow_core_framework_function_proto_depIdxs, MessageInfos: file_tensorflow_core_framework_function_proto_msgTypes, }.Build() File_tensorflow_core_framework_function_proto = out.File file_tensorflow_core_framework_function_proto_rawDesc = nil file_tensorflow_core_framework_function_proto_goTypes = nil file_tensorflow_core_framework_function_proto_depIdxs = nil }