# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: octoml_profile/protos/fxgraph.proto
"""Generated protocol buffer code."""
from google.protobuf.internal import enum_type_wrapper
from google.protobuf import descriptor as _descriptor
from google.protobuf import descriptor_pool as _descriptor_pool
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()




DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#octoml_profile/protos/fxgraph.proto\x12\x0eoctoml_fxproto\"\x11\n\x0fPlaceholderNode\"1\n\tListValue\x12$\n\x05items\x18\x01 \x03(\x0b\x32\x15.octoml_fxproto.Value\">\n\x0b\x44\x65viceValue\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x12\n\nwith_index\x18\x02 \x01(\x08\x12\r\n\x05index\x18\x03 \x01(\x03\"\x92\x01\n\x0bTensorValue\x12)\n\x05\x64type\x18\x01 \x01(\x0e\x32\x1a.octoml_fxproto.DtypeValue\x12\r\n\x05shape\x18\x02 \x03(\x03\x12\x19\n\x11weight_file_index\x18\x03 \x01(\x03\x12\x1a\n\x12weight_file_offset\x18\x04 \x01(\x03\x12\x12\n\nsize_bytes\x18\x05 \x01(\x03\"\xcb\x03\n\x05Value\x12\x13\n\tnode_name\x18\x01 \x01(\tH\x00\x12-\n\x06tensor\x18\x02 \x01(\x0b\x32\x1b.octoml_fxproto.TensorValueH\x00\x12\x11\n\x07integer\x18\x03 \x01(\tH\x00\x12\x11\n\x07\x62oolean\x18\x04 \x01(\x08H\x00\x12\x0f\n\x05\x66loat\x18\x05 \x01(\x01H\x00\x12\x0e\n\x04none\x18\x06 \x01(\x08H\x00\x12)\n\x04list\x18\x07 \x01(\x0b\x32\x19.octoml_fxproto.ListValueH\x00\x12*\n\x05tuple\x18\x08 \x01(\x0b\x32\x19.octoml_fxproto.ListValueH\x00\x12-\n\x06\x64\x65vice\x18\t \x01(\x0b\x32\x1b.octoml_fxproto.DeviceValueH\x00\x12+\n\x05\x64type\x18\n \x01(\x0e\x32\x1a.octoml_fxproto.DtypeValueH\x00\x12:\n\rmemory_format\x18\x0b \x01(\x0e\x32!.octoml_fxproto.MemoryFormatValueH\x00\x12-\n\x06layout\x18\x0c \x01(\x0e\x32\x1b.octoml_fxproto.LayoutValueH\x00\x12\x10\n\x06string\x18\r \x01(\tH\x00\x42\x07\n\x05value\"@\n\nKwArgValue\x12\x0c\n\x04name\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.octoml_fxproto.Value\"G\n\nOpOverload\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\x0f\n\x07op_name\x18\x02 \x01(\t\x12\x15\n\roverload_name\x18\x03 \x01(\t\"~\n\x08\x46unction\x12\x31\n\x0bop_overload\x18\x01 \x01(\x0b\x32\x1a.octoml_fxproto.OpOverloadH\x00\x12\x19\n\x0fpython_operator\x18\x02 \x01(\tH\x00\x12\x18\n\x0etorch_function\x18\x03 \x01(\tH\x00\x42\n\n\x08\x66unction\"\x8d\x01\n\x10\x43\x61llFunctionNode\x12(\n\x06target\x18\x01 \x01(\x0b\x32\x18.octoml_fxproto.Function\x12#\n\x04\x61rgs\x18\x02 \x03(\x0b\x32\x15.octoml_fxproto.Value\x12*\n\x06kwargs\x18\x03 \x03(\x0b\x32\x1a.octoml_fxproto.KwArgValue\"3\n\nOutputNode\x12%\n\x06result\x18\x01 \x01(\x0b\x32\x15.octoml_fxproto.Value\"\xbd\x01\n\x04Node\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x0bplaceholder\x18\x02 \x01(\x0b\x32\x1f.octoml_fxproto.PlaceholderNodeH\x00\x12\x39\n\rcall_function\x18\x03 \x01(\x0b\x32 .octoml_fxproto.CallFunctionNodeH\x00\x12,\n\x06output\x18\x04 \x01(\x0b\x32\x1a.octoml_fxproto.OutputNodeH\x00\x42\x06\n\x04kind\",\n\x05Graph\x12#\n\x05nodes\x18\x01 \x03(\x0b\x32\x14.octoml_fxproto.Node\"2\n\nFixedInput\x12$\n\x05value\x18\x01 \x01(\x0b\x32\x15.octoml_fxproto.Value\"_\n\x05Model\x12$\n\x05graph\x18\x01 \x01(\x0b\x32\x15.octoml_fxproto.Graph\x12\x30\n\x0c\x66ixed_inputs\x18\x02 \x03(\x0b\x32\x1a.octoml_fxproto.FixedInput*\x90\x02\n\nDtypeValue\x12\x11\n\rDTYPE_INVALID\x10\x00\x12\x0e\n\nDTYPE_BOOL\x10\x10\x12\x0e\n\nDTYPE_INT8\x10 \x12\x0f\n\x0b\x44TYPE_INT16\x10!\x12\x0f\n\x0b\x44TYPE_INT32\x10\"\x12\x0f\n\x0b\x44TYPE_INT64\x10#\x12\x0f\n\x0b\x44TYPE_UINT8\x10\x30\x12\x11\n\rDTYPE_FLOAT16\x10\x41\x12\x11\n\rDTYPE_FLOAT32\x10\x42\x12\x11\n\rDTYPE_FLOAT64\x10\x43\x12\x12\n\x0e\x44TYPE_BFLOAT16\x10Q\x12\x13\n\x0f\x44TYPE_COMPLEX32\x10\x62\x12\x13\n\x0f\x44TYPE_COMPLEX64\x10\x63\x12\x14\n\x10\x44TYPE_COMPLEX128\x10\x64*\x99\x01\n\x11MemoryFormatValue\x12\x1c\n\x18MEMORY_FORMAT_CONTIGUOUS\x10\x00\x12\x1f\n\x1bMEMORY_FORMAT_CHANNELS_LAST\x10\x01\x12\"\n\x1eMEMORY_FORMAT_CHANNELS_LAST_3D\x10\x02\x12!\n\x1dMEMORY_FORMAT_PRESERVE_FORMAT\x10\x03*8\n\x0bLayoutValue\x12\x12\n\x0eLAYOUT_STRIDED\x10\x00\x12\x15\n\x11LAYOUT_SPARSE_COO\x10\x01\x62\x06proto3')

_DTYPEVALUE = DESCRIPTOR.enum_types_by_name['DtypeValue']
DtypeValue = enum_type_wrapper.EnumTypeWrapper(_DTYPEVALUE)
_MEMORYFORMATVALUE = DESCRIPTOR.enum_types_by_name['MemoryFormatValue']
MemoryFormatValue = enum_type_wrapper.EnumTypeWrapper(_MEMORYFORMATVALUE)
_LAYOUTVALUE = DESCRIPTOR.enum_types_by_name['LayoutValue']
LayoutValue = enum_type_wrapper.EnumTypeWrapper(_LAYOUTVALUE)
DTYPE_INVALID = 0
DTYPE_BOOL = 16
DTYPE_INT8 = 32
DTYPE_INT16 = 33
DTYPE_INT32 = 34
DTYPE_INT64 = 35
DTYPE_UINT8 = 48
DTYPE_FLOAT16 = 65
DTYPE_FLOAT32 = 66
DTYPE_FLOAT64 = 67
DTYPE_BFLOAT16 = 81
DTYPE_COMPLEX32 = 98
DTYPE_COMPLEX64 = 99
DTYPE_COMPLEX128 = 100
MEMORY_FORMAT_CONTIGUOUS = 0
MEMORY_FORMAT_CHANNELS_LAST = 1
MEMORY_FORMAT_CHANNELS_LAST_3D = 2
MEMORY_FORMAT_PRESERVE_FORMAT = 3
LAYOUT_STRIDED = 0
LAYOUT_SPARSE_COO = 1


_PLACEHOLDERNODE = DESCRIPTOR.message_types_by_name['PlaceholderNode']
_LISTVALUE = DESCRIPTOR.message_types_by_name['ListValue']
_DEVICEVALUE = DESCRIPTOR.message_types_by_name['DeviceValue']
_TENSORVALUE = DESCRIPTOR.message_types_by_name['TensorValue']
_VALUE = DESCRIPTOR.message_types_by_name['Value']
_KWARGVALUE = DESCRIPTOR.message_types_by_name['KwArgValue']
_OPOVERLOAD = DESCRIPTOR.message_types_by_name['OpOverload']
_FUNCTION = DESCRIPTOR.message_types_by_name['Function']
_CALLFUNCTIONNODE = DESCRIPTOR.message_types_by_name['CallFunctionNode']
_OUTPUTNODE = DESCRIPTOR.message_types_by_name['OutputNode']
_NODE = DESCRIPTOR.message_types_by_name['Node']
_GRAPH = DESCRIPTOR.message_types_by_name['Graph']
_FIXEDINPUT = DESCRIPTOR.message_types_by_name['FixedInput']
_MODEL = DESCRIPTOR.message_types_by_name['Model']
PlaceholderNode = _reflection.GeneratedProtocolMessageType('PlaceholderNode', (_message.Message,), {
  'DESCRIPTOR' : _PLACEHOLDERNODE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.PlaceholderNode)
  })
_sym_db.RegisterMessage(PlaceholderNode)

ListValue = _reflection.GeneratedProtocolMessageType('ListValue', (_message.Message,), {
  'DESCRIPTOR' : _LISTVALUE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.ListValue)
  })
_sym_db.RegisterMessage(ListValue)

DeviceValue = _reflection.GeneratedProtocolMessageType('DeviceValue', (_message.Message,), {
  'DESCRIPTOR' : _DEVICEVALUE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.DeviceValue)
  })
_sym_db.RegisterMessage(DeviceValue)

TensorValue = _reflection.GeneratedProtocolMessageType('TensorValue', (_message.Message,), {
  'DESCRIPTOR' : _TENSORVALUE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.TensorValue)
  })
_sym_db.RegisterMessage(TensorValue)

Value = _reflection.GeneratedProtocolMessageType('Value', (_message.Message,), {
  'DESCRIPTOR' : _VALUE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.Value)
  })
_sym_db.RegisterMessage(Value)

KwArgValue = _reflection.GeneratedProtocolMessageType('KwArgValue', (_message.Message,), {
  'DESCRIPTOR' : _KWARGVALUE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.KwArgValue)
  })
_sym_db.RegisterMessage(KwArgValue)

OpOverload = _reflection.GeneratedProtocolMessageType('OpOverload', (_message.Message,), {
  'DESCRIPTOR' : _OPOVERLOAD,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.OpOverload)
  })
_sym_db.RegisterMessage(OpOverload)

Function = _reflection.GeneratedProtocolMessageType('Function', (_message.Message,), {
  'DESCRIPTOR' : _FUNCTION,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.Function)
  })
_sym_db.RegisterMessage(Function)

CallFunctionNode = _reflection.GeneratedProtocolMessageType('CallFunctionNode', (_message.Message,), {
  'DESCRIPTOR' : _CALLFUNCTIONNODE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.CallFunctionNode)
  })
_sym_db.RegisterMessage(CallFunctionNode)

OutputNode = _reflection.GeneratedProtocolMessageType('OutputNode', (_message.Message,), {
  'DESCRIPTOR' : _OUTPUTNODE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.OutputNode)
  })
_sym_db.RegisterMessage(OutputNode)

Node = _reflection.GeneratedProtocolMessageType('Node', (_message.Message,), {
  'DESCRIPTOR' : _NODE,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.Node)
  })
_sym_db.RegisterMessage(Node)

Graph = _reflection.GeneratedProtocolMessageType('Graph', (_message.Message,), {
  'DESCRIPTOR' : _GRAPH,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.Graph)
  })
_sym_db.RegisterMessage(Graph)

FixedInput = _reflection.GeneratedProtocolMessageType('FixedInput', (_message.Message,), {
  'DESCRIPTOR' : _FIXEDINPUT,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.FixedInput)
  })
_sym_db.RegisterMessage(FixedInput)

Model = _reflection.GeneratedProtocolMessageType('Model', (_message.Message,), {
  'DESCRIPTOR' : _MODEL,
  '__module__' : 'octoml_profile.protos.fxgraph_pb2'
  # @@protoc_insertion_point(class_scope:octoml_fxproto.Model)
  })
_sym_db.RegisterMessage(Model)

if _descriptor._USE_C_DESCRIPTORS == False:

  DESCRIPTOR._options = None
  _DTYPEVALUE._serialized_start=1652
  _DTYPEVALUE._serialized_end=1924
  _MEMORYFORMATVALUE._serialized_start=1927
  _MEMORYFORMATVALUE._serialized_end=2080
  _LAYOUTVALUE._serialized_start=2082
  _LAYOUTVALUE._serialized_end=2138
  _PLACEHOLDERNODE._serialized_start=55
  _PLACEHOLDERNODE._serialized_end=72
  _LISTVALUE._serialized_start=74
  _LISTVALUE._serialized_end=123
  _DEVICEVALUE._serialized_start=125
  _DEVICEVALUE._serialized_end=187
  _TENSORVALUE._serialized_start=190
  _TENSORVALUE._serialized_end=336
  _VALUE._serialized_start=339
  _VALUE._serialized_end=798
  _KWARGVALUE._serialized_start=800
  _KWARGVALUE._serialized_end=864
  _OPOVERLOAD._serialized_start=866
  _OPOVERLOAD._serialized_end=937
  _FUNCTION._serialized_start=939
  _FUNCTION._serialized_end=1065
  _CALLFUNCTIONNODE._serialized_start=1068
  _CALLFUNCTIONNODE._serialized_end=1209
  _OUTPUTNODE._serialized_start=1211
  _OUTPUTNODE._serialized_end=1262
  _NODE._serialized_start=1265
  _NODE._serialized_end=1454
  _GRAPH._serialized_start=1456
  _GRAPH._serialized_end=1500
  _FIXEDINPUT._serialized_start=1502
  _FIXEDINPUT._serialized_end=1552
  _MODEL._serialized_start=1554
  _MODEL._serialized_end=1649
# @@protoc_insertion_point(module_scope)
