gimlet-api 0.0.5__py3-none-any.whl → 0.0.7__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gimlet_api-0.0.5.dist-info → gimlet_api-0.0.7.dist-info}/METADATA +6 -3
- {gimlet_api-0.0.5.dist-info → gimlet_api-0.0.7.dist-info}/RECORD +24 -17
- gml/__init__.py +1 -0
- gml/asset_manager.py +75 -0
- gml/client.py +105 -19
- gml/compile.py +296 -34
- gml/device.py +77 -0
- gml/hf.py +524 -0
- gml/model.py +64 -18
- gml/model_utils.py +2 -0
- gml/pipelines.py +67 -40
- gml/preprocessing.py +5 -2
- gml/proto/src/api/corepb/v1/controlplane_pb2.py +7 -3
- gml/proto/src/api/corepb/v1/cp_edge_pb2.py +77 -63
- gml/proto/src/api/corepb/v1/device_info_pb2.py +43 -0
- gml/proto/src/api/corepb/v1/gem_config_pb2.py +45 -0
- gml/proto/src/api/corepb/v1/mediastream_pb2.py +33 -25
- gml/proto/src/api/corepb/v1/model_exec_pb2.py +105 -101
- gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py +40 -0
- gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2_grpc.py +66 -0
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py +7 -3
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py +33 -0
- gml/tensor.py +115 -34
- {gimlet_api-0.0.5.dist-info → gimlet_api-0.0.7.dist-info}/WHEEL +0 -0
@@ -22,7 +22,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
|
|
22
22
|
from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
|
23
23
|
|
24
24
|
|
25
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/model_exec.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a$mediapipe/framework/calculator.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xa7\x02\n\x0f\x42oundingBoxInfo\x12Z\n\nbox_format\x18\x01 \x01(\x0e\x32;.gml.internal.api.core.v1.BoundingBoxInfo.BoundingBoxFormatR\tboxFormat\x12%\n\x0e\x62ox_normalized\x18\x02 \x01(\x08R\rboxNormalized\"\x90\x01\n\x11\x42oundingBoxFormat\x12\x1f\n\x1b\x42OUNDING_BOX_FORMAT_UNKNOWN\x10\x00\x12\x1e\n\x1a\x42OUNDING_BOX_FORMAT_CXCYWH\x10\x01\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_YXYX\x10\x02\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_XYXY\x10\x03\"\xf4\x07\n\x16ImagePreprocessingStep\x12[\n\x04kind\x18\x01 \x01(\x0e\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImagePreprocessingKindR\x04kind\x12s\n\x11\x63onversion_params\x18\x02 \x01(\x0b\x32\x46.gml.internal.api.core.v1.ImagePreprocessingStep.ImageConversionParamsR\x10\x63onversionParams\x12g\n\rresize_params\x18\x03 \x01(\x0b\x32\x42.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParamsR\x0cresizeParams\x12v\n\x12standardize_params\x18\x04 \x01(\x0b\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImageStandardizeParamsR\x11standardizeParams\x1a-\n\x15ImageConversionParams\x12\x14\n\x05scale\x18\x01 \x01(\x08R\x05scale\x1a\xed\x01\n\x11ImageResizeParams\x12\x66\n\x04kind\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParams.ImageResizeKindR\x04kind\"p\n\x0fImageResizeKind\x12\x1d\n\x19IMAGE_RESIZE_KIND_UNKNOWN\x10\x00\x12\x1d\n\x19IMAGE_RESIZE_KIND_STRETCH\x10\x01\x12\x1f\n\x1bIMAGE_RESIZE_KIND_LETTERBOX\x10\x02\x1aH\n\x16ImageStandardizeParams\x12\x14\n\x05means\x18\x01 \x03(\x02R\x05means\x12\x18\n\x07stddevs\x18\x02 \x03(\x02R\x07stddevs\"\xbd\x01\n\x16ImagePreprocessingKind\x12$\n IMAGE_PREPROCESSING_KIND_UNKNOWN\x10\x00\x12.\n*IMAGE_PREPROCESSING_KIND_CONVERT_TO_TENSOR\x10\x01\x12#\n\x1fIMAGE_PREPROCESSING_KIND_RESIZE\x10\x02\x12(\n$IMAGE_PREPROCESSING_KIND_STANDARDIZE\x10\x03\"\xa1\x13\n\x12\x44imensionSemantics\x12W\n\x04kind\x18\x01 \x01(\x0e\x32\x43.gml.internal.api.core.v1.DimensionSemantics.DimensionSemanticsKindR\x04kind\x12q\n\x14image_channel_params\x18\x02 \x01(\x0b\x32?.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParamsR\x12imageChannelParams\x12\x86\x01\n\x1b\x64\x65tection_candidates_params\x18\x03 \x01(\x0b\x32\x46.gml.internal.api.core.v1.DimensionSemantics.DetectionCandidatesParamsR\x19\x64\x65tectionCandidatesParams\x12z\n\x17\x64\x65tection_output_params\x18\x04 \x01(\x0b\x32\x42.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParamsR\x15\x64\x65tectionOutputParams\x12}\n\x18segmentation_mask_params\x18\x05 \x01(\x0b\x32\x43.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParamsR\x16segmentationMaskParams\x12j\n\x11regression_params\x18\x06 \x01(\x0b\x32=.gml.internal.api.core.v1.DimensionSemantics.RegressionParamsR\x10regressionParams\x1a\xf4\x01\n\x12ImageChannelParams\x12j\n\x06\x66ormat\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParams.ImageChannelFormatR\x06\x66ormat\"r\n\x12ImageChannelFormat\x12 \n\x1cIMAGE_CHANNEL_FORMAT_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_RGB\x10\x01\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_BGR\x10\x02\x1a=\n\x19\x44\x65tectionCandidatesParams\x12 \n\x0cis_nms_boxes\x18\x01 \x01(\x08R\nisNmsBoxes\x1a\xdf\x03\n\x15\x44\x65tectionOutputParams\x12\x7f\n\x14\x62ox_coordinate_range\x18\x01 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x12\x62oxCoordinateRange\x12H\n\nbox_format\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\tboxFormat\x12\x30\n\x14\x62ox_confidence_index\x18\x03 \x01(\x05R\x12\x62oxConfidenceIndex\x12\x1f\n\x0b\x63lass_index\x18\x04 \x01(\x05R\nclassIndex\x12p\n\x0cscores_range\x18\x05 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x0bscoresRange\x1a\x36\n\nIndexRange\x12\x14\n\x05start\x18\x01 \x01(\x05R\x05start\x12\x12\n\x04size\x18\x02 \x01(\x05R\x04size\x1a\xae\x02\n\x16SegmentationMaskParams\x12l\n\x04kind\x18\x01 \x01(\x0e\x32X.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParams.SegmentationMaskKindR\x04kind\"\xa5\x01\n\x14SegmentationMaskKind\x12\"\n\x1eSEGMENTATION_MASK_KIND_UNKNOWN\x10\x00\x12&\n\"SEGMENTATION_MASK_KIND_CLASS_LABEL\x10\x01\x12\x1f\n\x1bSEGMENTATION_MASK_KIND_BOOL\x10\x02\x12 \n\x1cSEGMENTATION_MASK_KIND_SCORE\x10\x03\x1a\\\n\x10RegressionParams\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x32\n\x05scale\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x05scale\"\xa7\x04\n\x16\x44imensionSemanticsKind\x12$\n DIMENSION_SEMANTICS_KIND_UNKNOWN\x10\x00\x12\"\n\x1e\x44IMENSION_SEMANTICS_KIND_BATCH\x10\x01\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_IGNORE\x10\x02\x12*\n&DIMENSION_SEMANTICS_KIND_IMAGE_CHANNEL\x10\x03\x12)\n%DIMENSION_SEMANTICS_KIND_IMAGE_HEIGHT\x10\x04\x12(\n$DIMENSION_SEMANTICS_KIND_IMAGE_WIDTH\x10\x05\x12\x31\n-DIMENSION_SEMANTICS_KIND_DETECTION_CANDIDATES\x10\x06\x12-\n)DIMENSION_SEMANTICS_KIND_DETECTION_OUTPUT\x10\x07\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_SCORES\x10\x08\x12\x36\n2DIMENSION_SEMANTICS_KIND_SEGMENTATION_MASK_CHANNEL\x10\t\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_LABELS\x10\n\x12-\n)DIMENSION_SEMANTICS_KIND_REGRESSION_VALUE\x10\x0b\"_\n\x0fTensorSemantics\x12L\n\ndimensions\x18\x01 \x03(\x0b\x32,.gml.internal.api.core.v1.DimensionSemanticsR\ndimensions\"\xa7\t\n\tModelInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x04kind\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ModelInfo.ModelKindR\x04kind\x12N\n\x06\x66ormat\x18\x03 \x01(\x0e\x32\x36.gml.internal.api.core.v1.ModelInfo.ModelStorageFormatR\x06\x66ormat\x12T\n\x0b\x66ile_assets\x18\x04 \x03(\x0b\x32\x33.gml.internal.api.core.v1.ModelInfo.FileAssetsEntryR\nfileAssets\x12_\n\x16input_tensor_semantics\x18\x05 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x14inputTensorSemantics\x12\x61\n\x17output_tensor_semantics\x18\x06 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x15outputTensorSemantics\x12!\n\x0c\x63lass_labels\x18\x64 \x03(\tR\x0b\x63lassLabels\x12\x46\n\tbbox_info\x18\x65 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\x08\x62\x62oxInfo\x12l\n\x19image_preprocessing_steps\x18\x66 \x03(\x0b\x32\x30.gml.internal.api.core.v1.ImagePreprocessingStepR\x17imagePreprocessingSteps\x1aN\n\x0f\x46ileAssetsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDR\x05value:\x02\x38\x01\"\x9a\x01\n\tModelKind\x12\x16\n\x12MODEL_KIND_UNKNOWN\x10\x00\x12\x14\n\x10MODEL_KIND_TORCH\x10\x01\x12\x1a\n\x16MODEL_KIND_TORCHSCRIPT\x10\x02\x12\x13\n\x0fMODEL_KIND_ONNX\x10\x03\x12\x15\n\x11MODEL_KIND_TFLITE\x10\x04\x12\x17\n\x13MODEL_KIND_OPENVINO\x10\x05\"\x92\x02\n\x12ModelStorageFormat\x12 \n\x1cMODEL_STORAGE_FORMAT_UNKNOWN\x10\x00\x12&\n\"MODEL_STORAGE_FORMAT_MLIR_BYTECODE\x10\x01\x12\"\n\x1eMODEL_STORAGE_FORMAT_MLIR_TEXT\x10\x02\x12!\n\x1dMODEL_STORAGE_FORMAT_PROTOBUF\x10\x03\x12#\n\x1fMODEL_STORAGE_FORMAT_PROTO_TEXT\x10\x04\x12#\n\x1fMODEL_STORAGE_FORMAT_FLATBUFFER\x10\x05\x12!\n\x1dMODEL_STORAGE_FORMAT_OPENVINO\x10\x06\"\xa8\x02\n\x04Node\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x04kind\x18\x02 \x01(\x0e\x32\x31.gml.internal.api.core.v1.LogicalPipelineNodeKindR\x04kind\x12H\n\nattributes\x18\x03 \x03(\x0b\x32(.gml.internal.api.core.v1.NodeAttributesR\nattributes\x12;\n\x06inputs\x18\x04 \x03(\x0b\x32#.gml.internal.api.core.v1.NodeInputR\x06inputs\x12>\n\x07outputs\x18\x05 \x03(\x0b\x32$.gml.internal.api.core.v1.NodeOutputR\x07outputs\"\x1e\n\x08ParamRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\xab\x04\n\x05Value\x12!\n\x0bstring_data\x18\x01 \x01(\tH\x00R\nstringData\x12\x1f\n\nint64_data\x18\x02 \x01(\x03H\x00R\tint64Data\x12!\n\x0b\x64ouble_data\x18\x03 \x01(\x01H\x00R\ndoubleData\x12\x1d\n\tbool_data\x18\x04 \x01(\x08H\x00R\x08\x62oolData\x12I\n\x0blambda_data\x18\x05 \x01(\x0b\x32&.gml.internal.api.core.v1.Value.LambdaH\x00R\nlambdaData\x12I\n\nmodel_data\x18\x06 \x01(\x0b\x32(.gml.internal.api.core.v1.Value.ModelRefH\x00R\tmodelData\x12\x43\n\nparam_data\x18\x07 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\tparamData\x1aG\n\x08ModelRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x1ap\n\x06Lambda\x12\x16\n\x06inputs\x18\x01 \x03(\tR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12\x34\n\x05nodes\x18\x03 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodesB\x06\n\x04\x64\x61ta\"[\n\x0eNodeAttributes\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x05value\"\x9c\x03\n\tNodeInput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x0bparam_value\x18\x02 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\nparamValue\x12_\n\x11node_output_value\x18\x03 \x01(\x0b\x32\x31.gml.internal.api.core.v1.NodeInput.NodeOutputRefH\x00R\x0fnodeOutputValue\x12\x62\n\x12lambda_input_value\x18\x04 \x01(\x0b\x32\x32.gml.internal.api.core.v1.NodeInput.LambdaInputRefH\x00R\x10lambdaInputValue\x1a@\n\rNodeOutputRef\x12\x1b\n\tnode_name\x18\x01 \x01(\tR\x08nodeName\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x1a$\n\x0eLambdaInputRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04nameB\x07\n\x05value\" \n\nNodeOutput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"a\n\x05Param\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x44\n\rdefault_value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x0c\x64\x65\x66\x61ultValue\"\xbc\x01\n\x0fLogicalPipeline\x12\x37\n\x06params\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ParamR\x06params\x12\x34\n\x05nodes\x18\x02 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodes\x12:\n\tmodel_ids\x18\x03 \x03(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08ModelIDsR\x08modelIds\"\xab\x04\n\x12PipelineDeployment\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12V\n\x13logical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12\x37\n\x08\x66leet_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x18\n\x07version\x18\x06 \x01(\x03R\x07version\x12\x44\n\x04spec\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PipelineDeploymentSpecR\x04spec\x12J\n\x06status\x18\x08 \x01(\x0b\x32\x32.gml.internal.api.core.v1.PipelineDeploymentStatusR\x06status\x12\x39\n\ndeleted_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tdeletedAt\"W\n\x16PipelineDeploymentSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\"q\n\x18PipelineDeploymentStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\"\xdc\x03\n\x10PhysicalPipeline\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12_\n\x16pipeline_deployment_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12:\n\tdevice_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x04spec\x18\x06 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\x12H\n\x06status\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\xc8\x01\n\x14PhysicalPipelineSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12=\n\x05graph\x18\x03 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x18\n\x07runtime\x18\x04 \x01(\tR\x07runtime\"\xa3\x01\n\x16PhysicalPipelineStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12\x18\n\x07runtime\x18\x03 \x01(\tR\x07runtime\x12\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\"\x84\x01\n\x0c\x46ileResource\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12\x1d\n\nsize_bytes\x18\x02 \x01(\x04R\tsizeBytes\x12\x1f\n\x0bsha256_hash\x18\x03 \x01(\tR\nsha256Hash\"\x8b\x01\n\rExecutionSpec\x12\x36\n\x05graph\x18\x01 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x05graph\x12\x42\n\nmodel_spec\x18\x02 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelSpecR\tmodelSpec\"\\\n\nNamedAsset\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12:\n\x04\x66ile\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceR\x04\x66ile\"\xd0\x03\n\tModelSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x33\n\ronnx_blob_key\x18\x02 \x01(\tB\x0f\xe2\xde\x1f\x0bONNXBlobKeyR\x0bonnxBlobKey\x12Q\n\tonnx_file\x18\x03 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceB\x0c\xe2\xde\x1f\x08ONNXFileR\x08onnxFile\x12\x45\n\x0bnamed_asset\x18\x04 \x03(\x0b\x32$.gml.internal.api.core.v1.NamedAssetR\nnamedAsset\x12\x18\n\x07runtime\x18\x32 \x01(\tR\x07runtime\x12\x62\n\rtensorrt_spec\x18\x64 \x01(\x0b\x32+.gml.internal.api.core.v1.TensorRTModelSpecB\x10\xe2\xde\x1f\x0cTensorRTSpecR\x0ctensorrtSpec\x12\x62\n\ropenvino_spec\x18\x65 \x01(\x0b\x32+.gml.internal.api.core.v1.OpenVINOModelSpecB\x10\xe2\xde\x1f\x0cOpenVINOSpecR\x0copenvinoSpec\"\xfe\x01\n\x11TensorRTModelSpec\x12h\n\x14optimization_profile\x18\x01 \x03(\x0b\x32\x35.gml.internal.api.core.v1.TensorRTOptimizationProfileR\x13optimizationProfile\x12&\n\x0f\x65ngine_blob_key\x18\x02 \x01(\tR\rengineBlobKey\x12W\n\x0fmem_pool_limits\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.TensorRTMemPoolLimitsR\rmemPoolLimits\"\x7f\n\x1bTensorRTOptimizationProfile\x12`\n\x12tensor_shape_range\x18\x01 \x03(\x0b\x32\x32.gml.internal.api.core.v1.TensorRTTensorShapeRangeR\x10tensorShapeRange\"M\n\x18TensorRTTensorShapeRange\x12\x1f\n\x0btensor_name\x18\x01 \x01(\tR\ntensorName\x12\x10\n\x03\x64im\x18\x02 \x03(\x05R\x03\x64im\"5\n\x15TensorRTMemPoolLimits\x12\x1c\n\tworkspace\x18\x01 \x01(\x03R\tworkspace\"\x8e\x01\n\x11OpenVINOModelSpec\x12X\n\x0binput_shape\x18\x01 \x03(\x0b\x32\x37.gml.internal.api.core.v1.OpenVINOModelSpec.TensorShapeR\ninputShape\x1a\x1f\n\x0bTensorShape\x12\x10\n\x03\x64im\x18\x01 \x03(\x05R\x03\x64im*\xcc\x04\n\x17LogicalPipelineNodeKind\x12&\n\"LOGICAL_PIPELINE_NODE_KIND_UNKNOWN\x10\x00\x12,\n(LOGICAL_PIPELINE_NODE_KIND_CAMERA_SOURCE\x10\n\x12&\n!LOGICAL_PIPELINE_NODE_KIND_DETECT\x10\xe8\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_CLASSIFY\x10\xe9\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_SEGMENT\x10\xea\x07\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_MULTI_PURPOSE_MODEL\x10\xeb\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_REGRESS\x10\xec\x07\x12,\n\'LOGICAL_PIPELINE_NODE_KIND_FOR_EACH_ROI\x10\xdf\x0b\x12\x31\n,LOGICAL_PIPELINE_NODE_KIND_VIDEO_STREAM_SINK\x10\xd0\x0f\x12\x37\n2LOGICAL_PIPELINE_NODE_KIND_DETECTIONS_METRICS_SINK\x10\xd1\x0f\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_LATENCY_METRICS_SINK\x10\xd2\x0f\x12\x32\n-LOGICAL_PIPELINE_NODE_KIND_FRAME_METRICS_SINK\x10\xd3\x0f*\xd7\x01\n\rPipelineState\x12\x1a\n\x16PIPELINE_STATE_UNKNOWN\x10\x00\x12\x1a\n\x16PIPELINE_STATE_PENDING\x10\x01\x12\x18\n\x14PIPELINE_STATE_READY\x10\x02\x12\x1a\n\x16PIPELINE_STATE_RUNNING\x10\x03\x12\x1e\n\x1aPIPELINE_STATE_TERMINATING\x10\x04\x12\x1d\n\x19PIPELINE_STATE_TERMINATED\x10\x05\x12\x19\n\x15PIPELINE_STATE_FAILED\x10\x06\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
25
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/model_exec.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a$mediapipe/framework/calculator.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xa7\x02\n\x0f\x42oundingBoxInfo\x12Z\n\nbox_format\x18\x01 \x01(\x0e\x32;.gml.internal.api.core.v1.BoundingBoxInfo.BoundingBoxFormatR\tboxFormat\x12%\n\x0e\x62ox_normalized\x18\x02 \x01(\x08R\rboxNormalized\"\x90\x01\n\x11\x42oundingBoxFormat\x12\x1f\n\x1b\x42OUNDING_BOX_FORMAT_UNKNOWN\x10\x00\x12\x1e\n\x1a\x42OUNDING_BOX_FORMAT_CXCYWH\x10\x01\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_YXYX\x10\x02\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_XYXY\x10\x03\"\xb5\x08\n\x16ImagePreprocessingStep\x12[\n\x04kind\x18\x01 \x01(\x0e\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImagePreprocessingKindR\x04kind\x12s\n\x11\x63onversion_params\x18\x02 \x01(\x0b\x32\x46.gml.internal.api.core.v1.ImagePreprocessingStep.ImageConversionParamsR\x10\x63onversionParams\x12g\n\rresize_params\x18\x03 \x01(\x0b\x32\x42.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParamsR\x0cresizeParams\x12v\n\x12standardize_params\x18\x04 \x01(\x0b\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImageStandardizeParamsR\x11standardizeParams\x1an\n\x15ImageConversionParams\x12\x14\n\x05scale\x18\x01 \x01(\x08R\x05scale\x12?\n\x0cscale_factor\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x0bscaleFactor\x1a\xed\x01\n\x11ImageResizeParams\x12\x66\n\x04kind\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParams.ImageResizeKindR\x04kind\"p\n\x0fImageResizeKind\x12\x1d\n\x19IMAGE_RESIZE_KIND_UNKNOWN\x10\x00\x12\x1d\n\x19IMAGE_RESIZE_KIND_STRETCH\x10\x01\x12\x1f\n\x1bIMAGE_RESIZE_KIND_LETTERBOX\x10\x02\x1aH\n\x16ImageStandardizeParams\x12\x14\n\x05means\x18\x01 \x03(\x02R\x05means\x12\x18\n\x07stddevs\x18\x02 \x03(\x02R\x07stddevs\"\xbd\x01\n\x16ImagePreprocessingKind\x12$\n IMAGE_PREPROCESSING_KIND_UNKNOWN\x10\x00\x12.\n*IMAGE_PREPROCESSING_KIND_CONVERT_TO_TENSOR\x10\x01\x12#\n\x1fIMAGE_PREPROCESSING_KIND_RESIZE\x10\x02\x12(\n$IMAGE_PREPROCESSING_KIND_STANDARDIZE\x10\x03\"\x95\x15\n\x12\x44imensionSemantics\x12W\n\x04kind\x18\x01 \x01(\x0e\x32\x43.gml.internal.api.core.v1.DimensionSemantics.DimensionSemanticsKindR\x04kind\x12q\n\x14image_channel_params\x18\x02 \x01(\x0b\x32?.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParamsR\x12imageChannelParams\x12\x86\x01\n\x1b\x64\x65tection_candidates_params\x18\x03 \x01(\x0b\x32\x46.gml.internal.api.core.v1.DimensionSemantics.DetectionCandidatesParamsR\x19\x64\x65tectionCandidatesParams\x12z\n\x17\x64\x65tection_output_params\x18\x04 \x01(\x0b\x32\x42.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParamsR\x15\x64\x65tectionOutputParams\x12}\n\x18segmentation_mask_params\x18\x05 \x01(\x0b\x32\x43.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParamsR\x16segmentationMaskParams\x12j\n\x11regression_params\x18\x06 \x01(\x0b\x32=.gml.internal.api.core.v1.DimensionSemantics.RegressionParamsR\x10regressionParams\x1a\xf4\x01\n\x12ImageChannelParams\x12j\n\x06\x66ormat\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParams.ImageChannelFormatR\x06\x66ormat\"r\n\x12ImageChannelFormat\x12 \n\x1cIMAGE_CHANNEL_FORMAT_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_RGB\x10\x01\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_BGR\x10\x02\x1a=\n\x19\x44\x65tectionCandidatesParams\x12 \n\x0cis_nms_boxes\x18\x01 \x01(\x08R\nisNmsBoxes\x1a\x8b\x04\n\x15\x44\x65tectionOutputParams\x12\x7f\n\x14\x62ox_coordinate_range\x18\x01 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x12\x62oxCoordinateRange\x12H\n\nbox_format\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\tboxFormat\x12\x30\n\x14\x62ox_confidence_index\x18\x03 \x01(\x05R\x12\x62oxConfidenceIndex\x12\x1f\n\x0b\x63lass_index\x18\x04 \x01(\x05R\nclassIndex\x12p\n\x0cscores_range\x18\x05 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x0bscoresRange\x12*\n\x11scores_are_logits\x18\x06 \x01(\x08R\x0fscoresAreLogits\x1a\x36\n\nIndexRange\x12\x14\n\x05start\x18\x01 \x01(\x05R\x05start\x12\x12\n\x04size\x18\x02 \x01(\x05R\x04size\x1a\xd1\x02\n\x16SegmentationMaskParams\x12l\n\x04kind\x18\x01 \x01(\x0e\x32X.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParams.SegmentationMaskKindR\x04kind\"\xc8\x01\n\x14SegmentationMaskKind\x12\"\n\x1eSEGMENTATION_MASK_KIND_UNKNOWN\x10\x00\x12&\n\"SEGMENTATION_MASK_KIND_CLASS_LABEL\x10\x01\x12\x1f\n\x1bSEGMENTATION_MASK_KIND_BOOL\x10\x02\x12 \n\x1cSEGMENTATION_MASK_KIND_SCORE\x10\x03\x12!\n\x1dSEGMENTATION_MASK_KIND_LOGITS\x10\x04\x1a\\\n\x10RegressionParams\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x32\n\x05scale\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x05scale\"\xcc\x05\n\x16\x44imensionSemanticsKind\x12$\n DIMENSION_SEMANTICS_KIND_UNKNOWN\x10\x00\x12\"\n\x1e\x44IMENSION_SEMANTICS_KIND_BATCH\x10\x01\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_IGNORE\x10\x02\x12*\n&DIMENSION_SEMANTICS_KIND_IMAGE_CHANNEL\x10\x03\x12)\n%DIMENSION_SEMANTICS_KIND_IMAGE_HEIGHT\x10\x04\x12(\n$DIMENSION_SEMANTICS_KIND_IMAGE_WIDTH\x10\x05\x12\x31\n-DIMENSION_SEMANTICS_KIND_DETECTION_CANDIDATES\x10\x06\x12-\n)DIMENSION_SEMANTICS_KIND_DETECTION_OUTPUT\x10\x07\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_SCORES\x10\x08\x12\x36\n2DIMENSION_SEMANTICS_KIND_SEGMENTATION_MASK_CHANNEL\x10\t\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_LABELS\x10\n\x12-\n)DIMENSION_SEMANTICS_KIND_REGRESSION_VALUE\x10\x0b\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_TOKENS\x10\x0c\x12+\n\'DIMENSION_SEMANTICS_KIND_ATTENTION_MASK\x10\r\x12)\n%DIMENSION_SEMANTICS_KIND_VOCAB_LOGITS\x10\x0e\x12&\n\"DIMENSION_SEMANTICS_KIND_EMBEDDING\x10\x0f\"\xe9\x02\n\x0fTensorSemantics\x12L\n\ndimensions\x18\x01 \x03(\x0b\x32,.gml.internal.api.core.v1.DimensionSemanticsR\ndimensions\x12Q\n\x04kind\x18\x02 \x01(\x0e\x32=.gml.internal.api.core.v1.TensorSemantics.TensorSemanticsKindR\x04kind\"\xb4\x01\n\x13TensorSemanticsKind\x12!\n\x1dTENSOR_SEMANTICS_KIND_UNKNOWN\x10\x00\x12 \n\x1cTENSOR_SEMANTICS_KIND_UNUSED\x10\x01\x12#\n\x1fTENSOR_SEMANTICS_KIND_DIMENSION\x10\x02\x12\x33\n/TENSOR_SEMANTICS_KIND_ATTENTION_KEY_VALUE_CACHE\x10\x03\"6\n\x10GenerationConfig\x12\"\n\reos_token_ids\x18\x01 \x03(\x03R\x0b\x65osTokenIds\"\xc7\n\n\tModelInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x04kind\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ModelInfo.ModelKindR\x04kind\x12N\n\x06\x66ormat\x18\x03 \x01(\x0e\x32\x36.gml.internal.api.core.v1.ModelInfo.ModelStorageFormatR\x06\x66ormat\x12T\n\x0b\x66ile_assets\x18\x04 \x03(\x0b\x32\x33.gml.internal.api.core.v1.ModelInfo.FileAssetsEntryR\nfileAssets\x12_\n\x16input_tensor_semantics\x18\x05 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x14inputTensorSemantics\x12\x61\n\x17output_tensor_semantics\x18\x06 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x15outputTensorSemantics\x12!\n\x0c\x63lass_labels\x18\x64 \x03(\tR\x0b\x63lassLabels\x12\x46\n\tbbox_info\x18\x65 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\x08\x62\x62oxInfo\x12l\n\x19image_preprocessing_steps\x18\x66 \x03(\x0b\x32\x30.gml.internal.api.core.v1.ImagePreprocessingStepR\x17imagePreprocessingSteps\x12W\n\x11generation_config\x18g \x01(\x0b\x32*.gml.internal.api.core.v1.GenerationConfigR\x10generationConfig\x1aN\n\x0f\x46ileAssetsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDR\x05value:\x02\x38\x01\"\xc0\x01\n\tModelKind\x12\x16\n\x12MODEL_KIND_UNKNOWN\x10\x00\x12\x14\n\x10MODEL_KIND_TORCH\x10\x01\x12\x1a\n\x16MODEL_KIND_TORCHSCRIPT\x10\x02\x12\x13\n\x0fMODEL_KIND_ONNX\x10\x03\x12\x15\n\x11MODEL_KIND_TFLITE\x10\x04\x12\x17\n\x13MODEL_KIND_OPENVINO\x10\x05\x12$\n MODEL_KIND_HUGGINGFACE_TOKENIZER\x10\x06\"\xb3\x02\n\x12ModelStorageFormat\x12 \n\x1cMODEL_STORAGE_FORMAT_UNKNOWN\x10\x00\x12&\n\"MODEL_STORAGE_FORMAT_MLIR_BYTECODE\x10\x01\x12\"\n\x1eMODEL_STORAGE_FORMAT_MLIR_TEXT\x10\x02\x12!\n\x1dMODEL_STORAGE_FORMAT_PROTOBUF\x10\x03\x12#\n\x1fMODEL_STORAGE_FORMAT_PROTO_TEXT\x10\x04\x12#\n\x1fMODEL_STORAGE_FORMAT_FLATBUFFER\x10\x05\x12!\n\x1dMODEL_STORAGE_FORMAT_OPENVINO\x10\x06\x12\x1f\n\x1bMODEL_STORAGE_FORMAT_OPAQUE\x10\x07\"\xa8\x02\n\x04Node\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x04kind\x18\x02 \x01(\x0e\x32\x31.gml.internal.api.core.v1.LogicalPipelineNodeKindR\x04kind\x12H\n\nattributes\x18\x03 \x03(\x0b\x32(.gml.internal.api.core.v1.NodeAttributesR\nattributes\x12;\n\x06inputs\x18\x04 \x03(\x0b\x32#.gml.internal.api.core.v1.NodeInputR\x06inputs\x12>\n\x07outputs\x18\x05 \x03(\x0b\x32$.gml.internal.api.core.v1.NodeOutputR\x07outputs\"\x1e\n\x08ParamRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\xab\x04\n\x05Value\x12!\n\x0bstring_data\x18\x01 \x01(\tH\x00R\nstringData\x12\x1f\n\nint64_data\x18\x02 \x01(\x03H\x00R\tint64Data\x12!\n\x0b\x64ouble_data\x18\x03 \x01(\x01H\x00R\ndoubleData\x12\x1d\n\tbool_data\x18\x04 \x01(\x08H\x00R\x08\x62oolData\x12I\n\x0blambda_data\x18\x05 \x01(\x0b\x32&.gml.internal.api.core.v1.Value.LambdaH\x00R\nlambdaData\x12I\n\nmodel_data\x18\x06 \x01(\x0b\x32(.gml.internal.api.core.v1.Value.ModelRefH\x00R\tmodelData\x12\x43\n\nparam_data\x18\x07 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\tparamData\x1aG\n\x08ModelRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x1ap\n\x06Lambda\x12\x16\n\x06inputs\x18\x01 \x03(\tR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12\x34\n\x05nodes\x18\x03 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodesB\x06\n\x04\x64\x61ta\"[\n\x0eNodeAttributes\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x05value\"\x9c\x03\n\tNodeInput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x0bparam_value\x18\x02 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\nparamValue\x12_\n\x11node_output_value\x18\x03 \x01(\x0b\x32\x31.gml.internal.api.core.v1.NodeInput.NodeOutputRefH\x00R\x0fnodeOutputValue\x12\x62\n\x12lambda_input_value\x18\x04 \x01(\x0b\x32\x32.gml.internal.api.core.v1.NodeInput.LambdaInputRefH\x00R\x10lambdaInputValue\x1a@\n\rNodeOutputRef\x12\x1b\n\tnode_name\x18\x01 \x01(\tR\x08nodeName\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x1a$\n\x0eLambdaInputRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04nameB\x07\n\x05value\" \n\nNodeOutput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"a\n\x05Param\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x44\n\rdefault_value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x0c\x64\x65\x66\x61ultValue\"\xbc\x01\n\x0fLogicalPipeline\x12\x37\n\x06params\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ParamR\x06params\x12\x34\n\x05nodes\x18\x02 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodes\x12:\n\tmodel_ids\x18\x03 \x03(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08ModelIDsR\x08modelIds\"\xab\x04\n\x12PipelineDeployment\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12V\n\x13logical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12\x37\n\x08\x66leet_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x18\n\x07version\x18\x06 \x01(\x03R\x07version\x12\x44\n\x04spec\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PipelineDeploymentSpecR\x04spec\x12J\n\x06status\x18\x08 \x01(\x0b\x32\x32.gml.internal.api.core.v1.PipelineDeploymentStatusR\x06status\x12\x39\n\ndeleted_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tdeletedAt\"W\n\x16PipelineDeploymentSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\"q\n\x18PipelineDeploymentStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\"\xdc\x03\n\x10PhysicalPipeline\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12_\n\x16pipeline_deployment_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12:\n\tdevice_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x04spec\x18\x06 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\x12H\n\x06status\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\xfa\x01\n\x14PhysicalPipelineSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12=\n\x05graph\x18\x03 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x18\n\x07runtime\x18\x04 \x01(\tR\x07runtime\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\"\xd5\x01\n\x16PhysicalPipelineStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12\x18\n\x07runtime\x18\x03 \x01(\tR\x07runtime\x12\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\"\x84\x01\n\x0c\x46ileResource\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12\x1d\n\nsize_bytes\x18\x02 \x01(\x04R\tsizeBytes\x12\x1f\n\x0bsha256_hash\x18\x03 \x01(\tR\nsha256Hash\"\x8b\x01\n\rExecutionSpec\x12\x36\n\x05graph\x18\x01 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x05graph\x12\x42\n\nmodel_spec\x18\x02 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelSpecR\tmodelSpec\"\\\n\nNamedAsset\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12:\n\x04\x66ile\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceR\x04\x66ile\"\xd0\x03\n\tModelSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x33\n\ronnx_blob_key\x18\x02 \x01(\tB\x0f\xe2\xde\x1f\x0bONNXBlobKeyR\x0bonnxBlobKey\x12Q\n\tonnx_file\x18\x03 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceB\x0c\xe2\xde\x1f\x08ONNXFileR\x08onnxFile\x12\x45\n\x0bnamed_asset\x18\x04 \x03(\x0b\x32$.gml.internal.api.core.v1.NamedAssetR\nnamedAsset\x12\x18\n\x07runtime\x18\x32 \x01(\tR\x07runtime\x12\x62\n\rtensorrt_spec\x18\x64 \x01(\x0b\x32+.gml.internal.api.core.v1.TensorRTModelSpecB\x10\xe2\xde\x1f\x0cTensorRTSpecR\x0ctensorrtSpec\x12\x62\n\ropenvino_spec\x18\x65 \x01(\x0b\x32+.gml.internal.api.core.v1.OpenVINOModelSpecB\x10\xe2\xde\x1f\x0cOpenVINOSpecR\x0copenvinoSpec\"\xfe\x01\n\x11TensorRTModelSpec\x12h\n\x14optimization_profile\x18\x01 \x03(\x0b\x32\x35.gml.internal.api.core.v1.TensorRTOptimizationProfileR\x13optimizationProfile\x12&\n\x0f\x65ngine_blob_key\x18\x02 \x01(\tR\rengineBlobKey\x12W\n\x0fmem_pool_limits\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.TensorRTMemPoolLimitsR\rmemPoolLimits\"\x7f\n\x1bTensorRTOptimizationProfile\x12`\n\x12tensor_shape_range\x18\x01 \x03(\x0b\x32\x32.gml.internal.api.core.v1.TensorRTTensorShapeRangeR\x10tensorShapeRange\"M\n\x18TensorRTTensorShapeRange\x12\x1f\n\x0btensor_name\x18\x01 \x01(\tR\ntensorName\x12\x10\n\x03\x64im\x18\x02 \x03(\x05R\x03\x64im\"5\n\x15TensorRTMemPoolLimits\x12\x1c\n\tworkspace\x18\x01 \x01(\x03R\tworkspace\"\x8e\x01\n\x11OpenVINOModelSpec\x12X\n\x0binput_shape\x18\x01 \x03(\x0b\x32\x37.gml.internal.api.core.v1.OpenVINOModelSpec.TensorShapeR\ninputShape\x1a\x1f\n\x0bTensorShape\x12\x10\n\x03\x64im\x18\x01 \x03(\x05R\x03\x64im*\xec\x07\n\x17LogicalPipelineNodeKind\x12&\n\"LOGICAL_PIPELINE_NODE_KIND_UNKNOWN\x10\x00\x12,\n(LOGICAL_PIPELINE_NODE_KIND_CAMERA_SOURCE\x10\n\x12\x31\n-LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SOURCE\x10\x0b\x12&\n!LOGICAL_PIPELINE_NODE_KIND_DETECT\x10\xe8\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_CLASSIFY\x10\xe9\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_SEGMENT\x10\xea\x07\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_MULTI_PURPOSE_MODEL\x10\xeb\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_REGRESS\x10\xec\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_TRACK\x10\xed\x07\x12/\n*LOGICAL_PIPELINE_NODE_KIND_GENERATE_TOKENS\x10\xee\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_TOKENIZE\x10\xef\x07\x12*\n%LOGICAL_PIPELINE_NODE_KIND_DETOKENIZE\x10\xf0\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_EMBED\x10\xf1\x07\x12\x35\n0LOGICAL_PIPELINE_NODE_KIND_TEMPLATE_CHAT_MESSAGE\x10\xf2\x07\x12,\n\'LOGICAL_PIPELINE_NODE_KIND_FOR_EACH_ROI\x10\xdf\x0b\x12-\n(LOGICAL_PIPELINE_NODE_KIND_VECTOR_SEARCH\x10\xe0\x0b\x12\x31\n,LOGICAL_PIPELINE_NODE_KIND_VIDEO_STREAM_SINK\x10\xd0\x0f\x12\x37\n2LOGICAL_PIPELINE_NODE_KIND_DETECTIONS_METRICS_SINK\x10\xd1\x0f\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_LATENCY_METRICS_SINK\x10\xd2\x0f\x12\x32\n-LOGICAL_PIPELINE_NODE_KIND_FRAME_METRICS_SINK\x10\xd3\x0f\x12\x30\n+LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SINK\x10\xd4\x0f*\xd7\x01\n\rPipelineState\x12\x1a\n\x16PIPELINE_STATE_UNKNOWN\x10\x00\x12\x1a\n\x16PIPELINE_STATE_PENDING\x10\x01\x12\x18\n\x14PIPELINE_STATE_READY\x10\x02\x12\x1a\n\x16PIPELINE_STATE_RUNNING\x10\x03\x12\x1e\n\x1aPIPELINE_STATE_TERMINATING\x10\x04\x12\x1d\n\x19PIPELINE_STATE_TERMINATED\x10\x05\x12\x19\n\x15PIPELINE_STATE_FAILED\x10\x06\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
26
26
|
|
27
27
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
28
28
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.model_exec_pb2', globals())
|
@@ -58,110 +58,114 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
58
58
|
_MODELSPEC.fields_by_name['tensorrt_spec']._serialized_options = b'\342\336\037\014TensorRTSpec'
|
59
59
|
_MODELSPEC.fields_by_name['openvino_spec']._options = None
|
60
60
|
_MODELSPEC.fields_by_name['openvino_spec']._serialized_options = b'\342\336\037\014OpenVINOSpec'
|
61
|
-
_LOGICALPIPELINENODEKIND._serialized_start=
|
62
|
-
_LOGICALPIPELINENODEKIND._serialized_end=
|
63
|
-
_PIPELINESTATE._serialized_start=
|
64
|
-
_PIPELINESTATE._serialized_end=
|
61
|
+
_LOGICALPIPELINENODEKIND._serialized_start=11019
|
62
|
+
_LOGICALPIPELINENODEKIND._serialized_end=12023
|
63
|
+
_PIPELINESTATE._serialized_start=12026
|
64
|
+
_PIPELINESTATE._serialized_end=12241
|
65
65
|
_BOUNDINGBOXINFO._serialized_start=221
|
66
66
|
_BOUNDINGBOXINFO._serialized_end=516
|
67
67
|
_BOUNDINGBOXINFO_BOUNDINGBOXFORMAT._serialized_start=372
|
68
68
|
_BOUNDINGBOXINFO_BOUNDINGBOXFORMAT._serialized_end=516
|
69
69
|
_IMAGEPREPROCESSINGSTEP._serialized_start=519
|
70
|
-
_IMAGEPREPROCESSINGSTEP._serialized_end=
|
70
|
+
_IMAGEPREPROCESSINGSTEP._serialized_end=1596
|
71
71
|
_IMAGEPREPROCESSINGSTEP_IMAGECONVERSIONPARAMS._serialized_start=980
|
72
|
-
_IMAGEPREPROCESSINGSTEP_IMAGECONVERSIONPARAMS._serialized_end=
|
73
|
-
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS._serialized_start=
|
74
|
-
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS._serialized_end=
|
75
|
-
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS_IMAGERESIZEKIND._serialized_start=
|
76
|
-
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS_IMAGERESIZEKIND._serialized_end=
|
77
|
-
_IMAGEPREPROCESSINGSTEP_IMAGESTANDARDIZEPARAMS._serialized_start=
|
78
|
-
_IMAGEPREPROCESSINGSTEP_IMAGESTANDARDIZEPARAMS._serialized_end=
|
79
|
-
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_start=
|
80
|
-
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_end=
|
81
|
-
_DIMENSIONSEMANTICS._serialized_start=
|
82
|
-
_DIMENSIONSEMANTICS._serialized_end=
|
83
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_start=
|
84
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_end=
|
85
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_start=
|
86
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_end=
|
87
|
-
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_start=
|
88
|
-
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_end=
|
89
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_start=
|
90
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_end=
|
91
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_start=
|
92
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_end=
|
93
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_start=
|
94
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_end=
|
95
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_start=
|
96
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_end=
|
97
|
-
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_start=
|
98
|
-
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_end=
|
99
|
-
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_start=
|
100
|
-
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=
|
101
|
-
_TENSORSEMANTICS._serialized_start=
|
102
|
-
_TENSORSEMANTICS._serialized_end=
|
103
|
-
|
104
|
-
|
105
|
-
|
106
|
-
|
107
|
-
|
108
|
-
|
109
|
-
|
110
|
-
|
111
|
-
|
112
|
-
|
113
|
-
|
114
|
-
|
115
|
-
|
116
|
-
|
117
|
-
|
118
|
-
|
119
|
-
|
120
|
-
|
121
|
-
|
122
|
-
|
123
|
-
|
124
|
-
|
125
|
-
|
126
|
-
|
127
|
-
|
128
|
-
|
129
|
-
|
130
|
-
|
131
|
-
|
132
|
-
|
133
|
-
|
134
|
-
|
135
|
-
|
136
|
-
|
137
|
-
|
138
|
-
|
139
|
-
|
140
|
-
|
141
|
-
|
142
|
-
|
143
|
-
|
144
|
-
|
145
|
-
|
146
|
-
|
147
|
-
|
148
|
-
|
149
|
-
|
150
|
-
|
151
|
-
|
152
|
-
|
153
|
-
|
154
|
-
|
155
|
-
|
156
|
-
|
157
|
-
|
158
|
-
|
159
|
-
|
160
|
-
|
161
|
-
|
162
|
-
|
163
|
-
|
164
|
-
|
165
|
-
|
166
|
-
|
72
|
+
_IMAGEPREPROCESSINGSTEP_IMAGECONVERSIONPARAMS._serialized_end=1090
|
73
|
+
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS._serialized_start=1093
|
74
|
+
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS._serialized_end=1330
|
75
|
+
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS_IMAGERESIZEKIND._serialized_start=1218
|
76
|
+
_IMAGEPREPROCESSINGSTEP_IMAGERESIZEPARAMS_IMAGERESIZEKIND._serialized_end=1330
|
77
|
+
_IMAGEPREPROCESSINGSTEP_IMAGESTANDARDIZEPARAMS._serialized_start=1332
|
78
|
+
_IMAGEPREPROCESSINGSTEP_IMAGESTANDARDIZEPARAMS._serialized_end=1404
|
79
|
+
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_start=1407
|
80
|
+
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_end=1596
|
81
|
+
_DIMENSIONSEMANTICS._serialized_start=1599
|
82
|
+
_DIMENSIONSEMANTICS._serialized_end=4308
|
83
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_start=2322
|
84
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_end=2566
|
85
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_start=2452
|
86
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_end=2566
|
87
|
+
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_start=2568
|
88
|
+
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_end=2629
|
89
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_start=2632
|
90
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_end=3155
|
91
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_start=3101
|
92
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_end=3155
|
93
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_start=3158
|
94
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_end=3495
|
95
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_start=3295
|
96
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_end=3495
|
97
|
+
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_start=3497
|
98
|
+
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_end=3589
|
99
|
+
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_start=3592
|
100
|
+
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=4308
|
101
|
+
_TENSORSEMANTICS._serialized_start=4311
|
102
|
+
_TENSORSEMANTICS._serialized_end=4672
|
103
|
+
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_start=4492
|
104
|
+
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_end=4672
|
105
|
+
_GENERATIONCONFIG._serialized_start=4674
|
106
|
+
_GENERATIONCONFIG._serialized_end=4728
|
107
|
+
_MODELINFO._serialized_start=4731
|
108
|
+
_MODELINFO._serialized_end=6082
|
109
|
+
_MODELINFO_FILEASSETSENTRY._serialized_start=5499
|
110
|
+
_MODELINFO_FILEASSETSENTRY._serialized_end=5577
|
111
|
+
_MODELINFO_MODELKIND._serialized_start=5580
|
112
|
+
_MODELINFO_MODELKIND._serialized_end=5772
|
113
|
+
_MODELINFO_MODELSTORAGEFORMAT._serialized_start=5775
|
114
|
+
_MODELINFO_MODELSTORAGEFORMAT._serialized_end=6082
|
115
|
+
_NODE._serialized_start=6085
|
116
|
+
_NODE._serialized_end=6381
|
117
|
+
_PARAMREF._serialized_start=6383
|
118
|
+
_PARAMREF._serialized_end=6413
|
119
|
+
_VALUE._serialized_start=6416
|
120
|
+
_VALUE._serialized_end=6971
|
121
|
+
_VALUE_MODELREF._serialized_start=6778
|
122
|
+
_VALUE_MODELREF._serialized_end=6849
|
123
|
+
_VALUE_LAMBDA._serialized_start=6851
|
124
|
+
_VALUE_LAMBDA._serialized_end=6963
|
125
|
+
_NODEATTRIBUTES._serialized_start=6973
|
126
|
+
_NODEATTRIBUTES._serialized_end=7064
|
127
|
+
_NODEINPUT._serialized_start=7067
|
128
|
+
_NODEINPUT._serialized_end=7479
|
129
|
+
_NODEINPUT_NODEOUTPUTREF._serialized_start=7368
|
130
|
+
_NODEINPUT_NODEOUTPUTREF._serialized_end=7432
|
131
|
+
_NODEINPUT_LAMBDAINPUTREF._serialized_start=7434
|
132
|
+
_NODEINPUT_LAMBDAINPUTREF._serialized_end=7470
|
133
|
+
_NODEOUTPUT._serialized_start=7481
|
134
|
+
_NODEOUTPUT._serialized_end=7513
|
135
|
+
_PARAM._serialized_start=7515
|
136
|
+
_PARAM._serialized_end=7612
|
137
|
+
_LOGICALPIPELINE._serialized_start=7615
|
138
|
+
_LOGICALPIPELINE._serialized_end=7803
|
139
|
+
_PIPELINEDEPLOYMENT._serialized_start=7806
|
140
|
+
_PIPELINEDEPLOYMENT._serialized_end=8361
|
141
|
+
_PIPELINEDEPLOYMENTSPEC._serialized_start=8363
|
142
|
+
_PIPELINEDEPLOYMENTSPEC._serialized_end=8450
|
143
|
+
_PIPELINEDEPLOYMENTSTATUS._serialized_start=8452
|
144
|
+
_PIPELINEDEPLOYMENTSTATUS._serialized_end=8565
|
145
|
+
_PHYSICALPIPELINE._serialized_start=8568
|
146
|
+
_PHYSICALPIPELINE._serialized_end=9044
|
147
|
+
_PHYSICALPIPELINESPEC._serialized_start=9047
|
148
|
+
_PHYSICALPIPELINESPEC._serialized_end=9297
|
149
|
+
_PHYSICALPIPELINESTATUS._serialized_start=9300
|
150
|
+
_PHYSICALPIPELINESTATUS._serialized_end=9513
|
151
|
+
_FILERESOURCE._serialized_start=9516
|
152
|
+
_FILERESOURCE._serialized_end=9648
|
153
|
+
_EXECUTIONSPEC._serialized_start=9651
|
154
|
+
_EXECUTIONSPEC._serialized_end=9790
|
155
|
+
_NAMEDASSET._serialized_start=9792
|
156
|
+
_NAMEDASSET._serialized_end=9884
|
157
|
+
_MODELSPEC._serialized_start=9887
|
158
|
+
_MODELSPEC._serialized_end=10351
|
159
|
+
_TENSORRTMODELSPEC._serialized_start=10354
|
160
|
+
_TENSORRTMODELSPEC._serialized_end=10608
|
161
|
+
_TENSORRTOPTIMIZATIONPROFILE._serialized_start=10610
|
162
|
+
_TENSORRTOPTIMIZATIONPROFILE._serialized_end=10737
|
163
|
+
_TENSORRTTENSORSHAPERANGE._serialized_start=10739
|
164
|
+
_TENSORRTTENSORSHAPERANGE._serialized_end=10816
|
165
|
+
_TENSORRTMEMPOOLLIMITS._serialized_start=10818
|
166
|
+
_TENSORRTMEMPOOLLIMITS._serialized_end=10871
|
167
|
+
_OPENVINOMODELSPEC._serialized_start=10874
|
168
|
+
_OPENVINOMODELSPEC._serialized_end=11016
|
169
|
+
_OPENVINOMODELSPEC_TENSORSHAPE._serialized_start=10985
|
170
|
+
_OPENVINOMODELSPEC_TENSORSHAPE._serialized_end=11016
|
167
171
|
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,40 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: src/controlplane/compiler/cpb/v1/cpb.proto
|
4
|
+
"""Generated protocol buffer code."""
|
5
|
+
from google.protobuf.internal import builder as _builder
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
# @@protoc_insertion_point(imports)
|
10
|
+
|
11
|
+
_sym_db = _symbol_database.Default()
|
12
|
+
|
13
|
+
|
14
|
+
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
15
|
+
from gml.proto.src.api.corepb.v1 import cp_edge_pb2 as src_dot_api_dot_corepb_dot_v1_dot_cp__edge__pb2
|
16
|
+
from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_dot_uuid__pb2
|
17
|
+
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
18
|
+
|
19
|
+
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*src/controlplane/compiler/cpb/v1/cpb.proto\x12%gml.internal.controlplane.compiler.v1\x1a\"src/api/corepb/v1/model_exec.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"i\n\x05Model\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x37\n\x04info\x18\x02 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\x04info\"\xe3\x02\n\x0e\x43ompileRequest\x12V\n\x13logical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12]\n\x13\x64\x65vice_capabilities\x18\x03 \x01(\x0b\x32,.gml.internal.api.core.v1.DeviceCapabilitiesR\x12\x64\x65viceCapabilities\x12\x44\n\x06models\x18\x04 \x03(\x0b\x32,.gml.internal.controlplane.compiler.v1.ModelR\x06models\"W\n\x0f\x43ompileResponse\x12\x44\n\texec_spec\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x08\x65xecSpec2\x8b\x01\n\x0f\x43ompilerService\x12x\n\x07\x43ompile\x12\x35.gml.internal.controlplane.compiler.v1.CompileRequest\x1a\x36.gml.internal.controlplane.compiler.v1.CompileResponseB;Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpbb\x06proto3')
|
21
|
+
|
22
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
23
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.compiler.cpb.v1.cpb_pb2', globals())
|
24
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
25
|
+
|
26
|
+
DESCRIPTOR._options = None
|
27
|
+
DESCRIPTOR._serialized_options = b'Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpb'
|
28
|
+
_MODEL.fields_by_name['id']._options = None
|
29
|
+
_MODEL.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
30
|
+
_COMPILEREQUEST.fields_by_name['logical_pipeline_id']._options = None
|
31
|
+
_COMPILEREQUEST.fields_by_name['logical_pipeline_id']._serialized_options = b'\342\336\037\021LogicalPipelineID'
|
32
|
+
_MODEL._serialized_start=207
|
33
|
+
_MODEL._serialized_end=312
|
34
|
+
_COMPILEREQUEST._serialized_start=315
|
35
|
+
_COMPILEREQUEST._serialized_end=670
|
36
|
+
_COMPILERESPONSE._serialized_start=672
|
37
|
+
_COMPILERESPONSE._serialized_end=759
|
38
|
+
_COMPILERSERVICE._serialized_start=762
|
39
|
+
_COMPILERSERVICE._serialized_end=901
|
40
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,66 @@
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
3
|
+
import grpc
|
4
|
+
|
5
|
+
from gml.proto.src.controlplane.compiler.cpb.v1 import cpb_pb2 as src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2
|
6
|
+
|
7
|
+
|
8
|
+
class CompilerServiceStub(object):
|
9
|
+
"""Missing associated documentation comment in .proto file."""
|
10
|
+
|
11
|
+
def __init__(self, channel):
|
12
|
+
"""Constructor.
|
13
|
+
|
14
|
+
Args:
|
15
|
+
channel: A grpc.Channel.
|
16
|
+
"""
|
17
|
+
self.Compile = channel.unary_unary(
|
18
|
+
'/gml.internal.controlplane.compiler.v1.CompilerService/Compile',
|
19
|
+
request_serializer=src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileRequest.SerializeToString,
|
20
|
+
response_deserializer=src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileResponse.FromString,
|
21
|
+
)
|
22
|
+
|
23
|
+
|
24
|
+
class CompilerServiceServicer(object):
|
25
|
+
"""Missing associated documentation comment in .proto file."""
|
26
|
+
|
27
|
+
def Compile(self, request, context):
|
28
|
+
"""Missing associated documentation comment in .proto file."""
|
29
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
30
|
+
context.set_details('Method not implemented!')
|
31
|
+
raise NotImplementedError('Method not implemented!')
|
32
|
+
|
33
|
+
|
34
|
+
def add_CompilerServiceServicer_to_server(servicer, server):
|
35
|
+
rpc_method_handlers = {
|
36
|
+
'Compile': grpc.unary_unary_rpc_method_handler(
|
37
|
+
servicer.Compile,
|
38
|
+
request_deserializer=src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileRequest.FromString,
|
39
|
+
response_serializer=src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileResponse.SerializeToString,
|
40
|
+
),
|
41
|
+
}
|
42
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
43
|
+
'gml.internal.controlplane.compiler.v1.CompilerService', rpc_method_handlers)
|
44
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
45
|
+
|
46
|
+
|
47
|
+
# This class is part of an EXPERIMENTAL API.
|
48
|
+
class CompilerService(object):
|
49
|
+
"""Missing associated documentation comment in .proto file."""
|
50
|
+
|
51
|
+
@staticmethod
|
52
|
+
def Compile(request,
|
53
|
+
target,
|
54
|
+
options=(),
|
55
|
+
channel_credentials=None,
|
56
|
+
call_credentials=None,
|
57
|
+
insecure=False,
|
58
|
+
compression=None,
|
59
|
+
wait_for_ready=None,
|
60
|
+
timeout=None,
|
61
|
+
metadata=None):
|
62
|
+
return grpc.experimental.unary_unary(request, target, '/gml.internal.controlplane.compiler.v1.CompilerService/Compile',
|
63
|
+
src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileRequest.SerializeToString,
|
64
|
+
src_dot_controlplane_dot_compiler_dot_cpb_dot_v1_dot_cpb__pb2.CompileResponse.FromString,
|
65
|
+
options, channel_credentials,
|
66
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
@@ -16,7 +16,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
|
|
16
16
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
17
17
|
|
18
18
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"R\n\x13LogicalPipelineInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"P\n\x1bListLogicalPipelinesRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\x8b\x01\n\x19GetLogicalPipelineRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\xa5\x02\n\x1aGetLogicalPipelineResponse\x12\x66\n\rpipeline_info\x18\x02 \x01(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\x0cpipelineInfo\x12\x12\n\x04yaml\x18\x03 \x01(\tR\x04yaml\x12T\n\x10logical_pipeline\x18\x04 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12/\n\x0b\x65ncoded_dag\x18\x05 \x01(\tB\x0e\xe2\xde\x1f\nEncodedDAGR\nencodedDagJ\x04\x08\x01\x10\x02\"\x7f\n\x1c\x43reateLogicalPipelineRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yamlJ\x04\x08\x03\x10\x04\"H\n\x1d\x43reateLogicalPipelineResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"R\n\x13LogicalPipelineInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"P\n\x1bListLogicalPipelinesRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\x8b\x01\n\x19GetLogicalPipelineRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\xa5\x02\n\x1aGetLogicalPipelineResponse\x12\x66\n\rpipeline_info\x18\x02 \x01(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\x0cpipelineInfo\x12\x12\n\x04yaml\x18\x03 \x01(\tR\x04yaml\x12T\n\x10logical_pipeline\x18\x04 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12/\n\x0b\x65ncoded_dag\x18\x05 \x01(\tB\x0e\xe2\xde\x1f\nEncodedDAGR\nencodedDagJ\x04\x08\x01\x10\x02\"\x7f\n\x1c\x43reateLogicalPipelineRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yamlJ\x04\x08\x03\x10\x04\"H\n\x1d\x43reateLogicalPipelineResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"5\n\x1fParseLogicalPipelineYAMLRequest\x12\x12\n\x04yaml\x18\x01 \x01(\tR\x04yaml\"x\n ParseLogicalPipelineYAMLResponse\x12T\n\x10logical_pipeline\x18\x01 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline2\xe1\x05\n\x16LogicalPipelineService\x12\xad\x01\n\x14ListLogicalPipelines\x12I.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesRequest\x1aJ.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesResponse\x12\xa7\x01\n\x12GetLogicalPipeline\x12G.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineRequest\x1aH.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineResponse\x12\xb0\x01\n\x15\x43reateLogicalPipeline\x12J.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineRequest\x1aK.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineResponse\x12\xb9\x01\n\x18ParseLogicalPipelineYAML\x12M.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLRequest\x1aN.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLResponseBDZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppbb\x06proto3')
|
20
20
|
|
21
21
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
22
22
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.logicalpipeline.lppb.v1.lppb_pb2', globals())
|
@@ -52,6 +52,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
52
52
|
_CREATELOGICALPIPELINEREQUEST._serialized_end=1050
|
53
53
|
_CREATELOGICALPIPELINERESPONSE._serialized_start=1052
|
54
54
|
_CREATELOGICALPIPELINERESPONSE._serialized_end=1124
|
55
|
-
|
56
|
-
|
55
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=1126
|
56
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=1179
|
57
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=1181
|
58
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=1301
|
59
|
+
_LOGICALPIPELINESERVICE._serialized_start=1304
|
60
|
+
_LOGICALPIPELINESERVICE._serialized_end=2041
|
57
61
|
# @@protoc_insertion_point(module_scope)
|
@@ -29,6 +29,11 @@ class LogicalPipelineServiceStub(object):
|
|
29
29
|
request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineRequest.SerializeToString,
|
30
30
|
response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineResponse.FromString,
|
31
31
|
)
|
32
|
+
self.ParseLogicalPipelineYAML = channel.unary_unary(
|
33
|
+
'/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ParseLogicalPipelineYAML',
|
34
|
+
request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.SerializeToString,
|
35
|
+
response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
|
36
|
+
)
|
32
37
|
|
33
38
|
|
34
39
|
class LogicalPipelineServiceServicer(object):
|
@@ -52,6 +57,12 @@ class LogicalPipelineServiceServicer(object):
|
|
52
57
|
context.set_details('Method not implemented!')
|
53
58
|
raise NotImplementedError('Method not implemented!')
|
54
59
|
|
60
|
+
def ParseLogicalPipelineYAML(self, request, context):
|
61
|
+
"""Missing associated documentation comment in .proto file."""
|
62
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
63
|
+
context.set_details('Method not implemented!')
|
64
|
+
raise NotImplementedError('Method not implemented!')
|
65
|
+
|
55
66
|
|
56
67
|
def add_LogicalPipelineServiceServicer_to_server(servicer, server):
|
57
68
|
rpc_method_handlers = {
|
@@ -70,6 +81,11 @@ def add_LogicalPipelineServiceServicer_to_server(servicer, server):
|
|
70
81
|
request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineRequest.FromString,
|
71
82
|
response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineResponse.SerializeToString,
|
72
83
|
),
|
84
|
+
'ParseLogicalPipelineYAML': grpc.unary_unary_rpc_method_handler(
|
85
|
+
servicer.ParseLogicalPipelineYAML,
|
86
|
+
request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.FromString,
|
87
|
+
response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.SerializeToString,
|
88
|
+
),
|
73
89
|
}
|
74
90
|
generic_handler = grpc.method_handlers_generic_handler(
|
75
91
|
'gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService', rpc_method_handlers)
|
@@ -130,3 +146,20 @@ class LogicalPipelineService(object):
|
|
130
146
|
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineResponse.FromString,
|
131
147
|
options, channel_credentials,
|
132
148
|
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
149
|
+
|
150
|
+
@staticmethod
|
151
|
+
def ParseLogicalPipelineYAML(request,
|
152
|
+
target,
|
153
|
+
options=(),
|
154
|
+
channel_credentials=None,
|
155
|
+
call_credentials=None,
|
156
|
+
insecure=False,
|
157
|
+
compression=None,
|
158
|
+
wait_for_ready=None,
|
159
|
+
timeout=None,
|
160
|
+
metadata=None):
|
161
|
+
return grpc.experimental.unary_unary(request, target, '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ParseLogicalPipelineYAML',
|
162
|
+
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.SerializeToString,
|
163
|
+
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
|
164
|
+
options, channel_credentials,
|
165
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|