gimlet-api 0.0.8__py3-none-any.whl → 0.0.9__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.8.dist-info → gimlet_api-0.0.9.dist-info}/METADATA +1 -1
- {gimlet_api-0.0.8.dist-info → gimlet_api-0.0.9.dist-info}/RECORD +13 -13
- gml/client.py +2 -2
- gml/device.py +2 -0
- gml/hf.py +8 -0
- gml/pipelines.py +21 -1
- gml/proto/src/api/corepb/v1/controlplane_pb2.py +34 -29
- gml/proto/src/api/corepb/v1/device_info_pb2.py +13 -9
- gml/proto/src/api/corepb/v1/gem_config_pb2.py +24 -15
- gml/proto/src/api/corepb/v1/model_exec_pb2.py +101 -97
- gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py +10 -7
- gml/tensor.py +4 -0
- {gimlet_api-0.0.8.dist-info → gimlet_api-0.0.9.dist-info}/WHEEL +0 -0
@@ -1,13 +1,13 @@
|
|
1
1
|
gml/__init__.py,sha256=H3WQZ_RaN7VNeb__qeHEbKLEwkaG7gpL5FQ8s1IotUA,773
|
2
2
|
gml/_utils.py,sha256=mSCWHhCdzcUvHqmJIB2FS215K1LMgJCWcZ6e6FWK3hQ,1184
|
3
3
|
gml/asset_manager.py,sha256=VnbqUZHPOgPrAh6ri9C0EuNhS8tAHIrbUyJPAJuD9po,2053
|
4
|
-
gml/client.py,sha256=
|
4
|
+
gml/client.py,sha256=jPA71PTbv-4DX2FsfFcj1VPN-eMGdF4tKCj2NnOW7ZE,13862
|
5
5
|
gml/compile.py,sha256=Ih43r_zU07p91w9aiA0lrPJfmACpAWg0x_HFddMSy7Q,8346
|
6
|
-
gml/device.py,sha256=
|
7
|
-
gml/hf.py,sha256=
|
6
|
+
gml/device.py,sha256=Iw71NnuLcgjY32ZMXHlnlPkosTuHEmL9E98utmNChlM,2650
|
7
|
+
gml/hf.py,sha256=pp215wNmaPyCVy4DqFJbe_vEe1BRJ1GAJEURZnLuU0g,28220
|
8
8
|
gml/model.py,sha256=xESdD7tlqn93ym67Lyyk7TZdM3wUqyn7qWdP2AbgdkI,7261
|
9
9
|
gml/model_utils.py,sha256=vZvE5cHZIDkUkeZ4Pk4hhV-zOYMiREluv4b8kdqQ3Ig,1375
|
10
|
-
gml/pipelines.py,sha256=
|
10
|
+
gml/pipelines.py,sha256=LKj_lh5I5HzyUUIPG4CImiqBnQPrJsj0CHPKhLiOOGo,8374
|
11
11
|
gml/preprocessing.py,sha256=MaKkEW4ZP9fjpkJQfpc0X3rCUuSuSmJnGMClHamKmZU,3210
|
12
12
|
gml/proto/gogoproto/gogo_pb2.py,sha256=WVMIAR8K--mCUkTPM7mEeeXGpQlRRtt_kco10iP3CZs,15728
|
13
13
|
gml/proto/mediapipe/framework/calculator_contract_test_pb2.py,sha256=hNjyZCBz3RYa6rN4xR3FOCZKA24gq_LsJ3EMegl5wK4,2031
|
@@ -25,16 +25,16 @@ gml/proto/mediapipe/framework/thread_pool_executor_pb2.py,sha256=9TJ66fqSo1BiJmE
|
|
25
25
|
gml/proto/opentelemetry/proto/common/v1/common_pb2.py,sha256=wQjeDti-C8JiNwRn-z5M5p-Fqxm-SmnbPaoitJcSK-4,2860
|
26
26
|
gml/proto/opentelemetry/proto/metrics/v1/metrics_pb2.py,sha256=t2Far6oVcUFQIimzgAkZ8vQd0asMIlvECp4osC0ujgg,9735
|
27
27
|
gml/proto/opentelemetry/proto/resource/v1/resource_pb2.py,sha256=cbNmE12Nm3PjW4NXU7-Z-9m_0Zs3Ab8R1xLkDnvclCg,1730
|
28
|
-
gml/proto/src/api/corepb/v1/controlplane_pb2.py,sha256=
|
28
|
+
gml/proto/src/api/corepb/v1/controlplane_pb2.py,sha256=BosvQ6GYaUGsNTkRZH7osP2dZGWP6U9WyxItIQ_QS-8,9769
|
29
29
|
gml/proto/src/api/corepb/v1/cp_edge_pb2.py,sha256=oIpxq13C1ynK3alzDNZTOL5URxz5qzbDLD9NOM5xxjE,14511
|
30
|
-
gml/proto/src/api/corepb/v1/device_info_pb2.py,sha256
|
31
|
-
gml/proto/src/api/corepb/v1/gem_config_pb2.py,sha256=
|
30
|
+
gml/proto/src/api/corepb/v1/device_info_pb2.py,sha256=5orIOJAkvtH9pWBSXveDASFi4Rn59YWdOSnVLdj891A,5356
|
31
|
+
gml/proto/src/api/corepb/v1/gem_config_pb2.py,sha256=2ljfF16Xeqgj9TM3gHN54BqRHqS3SQNhOCenEY9K9qU,4718
|
32
32
|
gml/proto/src/api/corepb/v1/mediastream_pb2.py,sha256=Un9OwDUmWdqv92QP66K-WVOAzxP_4hMoz33JI4W1G5Y,7868
|
33
|
-
gml/proto/src/api/corepb/v1/model_exec_pb2.py,sha256=
|
33
|
+
gml/proto/src/api/corepb/v1/model_exec_pb2.py,sha256=o0drstrDssejqCFo8Cmm9F0zDw_bmzeOUHiYFrruOqE,29877
|
34
34
|
gml/proto/src/common/typespb/jwt_pb2.py,sha256=lxy-bqbyg96i9n_xr2JbkuWX-ldnoJavXPMnApzVSio,5580
|
35
35
|
gml/proto/src/common/typespb/status_pb2.py,sha256=IbBJnbsAlvsuTtyT285ZuW6k5VaPfl5kRSOnBxD_H8M,2109
|
36
36
|
gml/proto/src/common/typespb/uuid_pb2.py,sha256=5Fm3jYpCPX7sMrP6RhRYsF0SnuZNIBEQJk9f0jwZ2Rw,1188
|
37
|
-
gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py,sha256
|
37
|
+
gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py,sha256=R8jcxOlR1iz4Y7MnxIKoJ2RaNayqWPiBSt0W496QT-c,3262
|
38
38
|
gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2_grpc.py,sha256=l-gTK9nYpTlVb7QGAckSQXlHhkRdKe2-nrxXc8NQavY,2912
|
39
39
|
gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2.py,sha256=KgoUT8ccF-yJPe1r4otQjAPQoKBaQzdBlHoIUSkk0yE,11445
|
40
40
|
gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2_grpc.py,sha256=p3OpT8-hfNHu4-29qr-ZahRwO-LoCYM9Q4jomAHTXGA,24572
|
@@ -45,7 +45,7 @@ gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py,sha256=-snjW
|
|
45
45
|
gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py,sha256=RVedXkNYu2iF5OHiXoYyRw9AGRCUWG7qNyY-5QY71Go,3762
|
46
46
|
gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py,sha256=KSdb6V04qUHDsb1R2o3wixwTyZgrhwnPYobjnRgWX4I,4735
|
47
47
|
gml/register_submodules.py,sha256=U8IwjVygX2vxNi_aK6ljHOD4mmrOhbyVczvy4wwulqU,5027
|
48
|
-
gml/tensor.py,sha256=
|
49
|
-
gimlet_api-0.0.
|
50
|
-
gimlet_api-0.0.
|
51
|
-
gimlet_api-0.0.
|
48
|
+
gml/tensor.py,sha256=aPLm3I3qkYNDcJmntaUycqqN5rsZmcj8ql0EkupJudY,14977
|
49
|
+
gimlet_api-0.0.9.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
|
50
|
+
gimlet_api-0.0.9.dist-info/METADATA,sha256=P5wOKzPZyJroiZTPRpWsDdjiS5XQL21GK-heo5Set_E,531
|
51
|
+
gimlet_api-0.0.9.dist-info/RECORD,,
|
gml/client.py
CHANGED
@@ -351,8 +351,8 @@ class Client:
|
|
351
351
|
models: List[Model],
|
352
352
|
pipeline_file: Optional[Path] = None,
|
353
353
|
pipeline: Optional[Union[str, Pipeline]] = None,
|
354
|
-
runtimes
|
355
|
-
cameras
|
354
|
+
runtimes: List[str],
|
355
|
+
cameras: List[str],
|
356
356
|
):
|
357
357
|
model_with_assets = []
|
358
358
|
for model in models:
|
gml/device.py
CHANGED
@@ -57,6 +57,8 @@ def _runtime_str_to_runtime_protos(
|
|
57
57
|
return deviceinfopb.ModelRuntimeType.MODEL_RUNTIME_TYPE_TENSORRT
|
58
58
|
case "openvino":
|
59
59
|
return deviceinfopb.ModelRuntimeType.MODEL_RUNTIME_TYPE_OPENVINO
|
60
|
+
case "hailort":
|
61
|
+
return deviceinfopb.ModelRuntimeType.MODEL_RUNTIME_TYPE_HAILORT
|
60
62
|
case _:
|
61
63
|
raise ValueError("invalid runtime: {}".format(runtime))
|
62
64
|
|
gml/hf.py
CHANGED
@@ -334,6 +334,14 @@ class HuggingFaceImageProcessor:
|
|
334
334
|
raise NotImplementedError(
|
335
335
|
"only semantic segmentation is currently supported"
|
336
336
|
)
|
337
|
+
# TODO(philkuz): Support panoptic segmentation models. Multiple outputs come from panoptic segmentation models.
|
338
|
+
# We need to decide whether we should invest in converting the panoptic segmentation output to semantic segmentation
|
339
|
+
# format or if we should directly support panoptic segmentation output.
|
340
|
+
if hasattr(self.processor, "post_process_panoptic_segmentation"):
|
341
|
+
raise NotImplementedError(
|
342
|
+
"panoptic segmentation models are not supported yet"
|
343
|
+
)
|
344
|
+
|
337
345
|
dimensions = [
|
338
346
|
BatchDimension(),
|
339
347
|
# TODO(james): verify all semantic segmentation in hugging face output a logits mask.
|
gml/pipelines.py
CHANGED
@@ -173,6 +173,16 @@ nodes:
|
|
173
173
|
|
174
174
|
|
175
175
|
class LiveChatPipeline(Pipeline):
|
176
|
+
def __init__(
|
177
|
+
self,
|
178
|
+
message_template: str = r'''"{% for message in messages %}{% if message['role'] == 'system' %}{{message['content']}}{% else %}{{'<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n' + message['content'] + '<|eot_id|>'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|start_header_id|>assistant<|end_header_id|>\n' }}{% endif %}"''',
|
179
|
+
preset_system_prompt: str = r"'<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nYou are a helpful assistant<|eot_id|>'",
|
180
|
+
add_generation_prompt: bool = True,
|
181
|
+
):
|
182
|
+
self.add_generation_prompt = add_generation_prompt
|
183
|
+
self.message_template = message_template
|
184
|
+
self.preset_system_prompt = preset_system_prompt
|
185
|
+
|
176
186
|
def to_yaml(self, models: List[Model], org_name: str) -> str:
|
177
187
|
if len(models) != 2:
|
178
188
|
raise ValueError(
|
@@ -195,6 +205,16 @@ nodes:
|
|
195
205
|
kind: TextStreamSource
|
196
206
|
outputs:
|
197
207
|
- prompt
|
208
|
+
- name: query_template
|
209
|
+
kind: TemplateChatMessage
|
210
|
+
attributes:
|
211
|
+
add_generation_prompt: {self.add_generation_prompt}
|
212
|
+
message_template: {self.message_template}
|
213
|
+
preset_system_prompt: {self.preset_system_prompt}
|
214
|
+
inputs:
|
215
|
+
query: .text_source.prompt
|
216
|
+
outputs:
|
217
|
+
- chat_message
|
198
218
|
- name: tokenize
|
199
219
|
kind: Tokenize
|
200
220
|
attributes:
|
@@ -203,7 +223,7 @@ nodes:
|
|
203
223
|
name: {tokenizer.name}
|
204
224
|
org: {org_name}
|
205
225
|
inputs:
|
206
|
-
text: .
|
226
|
+
text: .query_template.chat_message
|
207
227
|
outputs:
|
208
228
|
- tokens
|
209
229
|
- name: generate
|
@@ -15,12 +15,13 @@ from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
|
15
15
|
from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
|
16
16
|
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
17
17
|
from gml.proto.src.api.corepb.v1 import cp_edge_pb2 as src_dot_api_dot_corepb_dot_v1_dot_cp__edge__pb2
|
18
|
+
from gml.proto.src.api.corepb.v1 import gem_config_pb2 as src_dot_api_dot_corepb_dot_v1_dot_gem__config__pb2
|
18
19
|
from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_dot_uuid__pb2
|
19
20
|
from gml.proto.src.common.typespb import status_pb2 as src_dot_common_dot_typespb_dot_status__pb2
|
20
21
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
21
22
|
|
22
23
|
|
23
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/api/corepb/v1/controlplane.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\xc4\x01\n\nCPMetadata\x12\x37\n\x05topic\x18\x01 \x01(\x0e\x32!.gml.internal.api.core.v1.CPTopicR\x05topic\x12:\n\tentity_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x45ntityIDR\x08\x65ntityId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"v\n\tCPMessage\x12@\n\x08metadata\x18\x01 \x01(\x0b\x32$.gml.internal.api.core.v1.CPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"M\n\x0f\x44\x65viceConnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"J\n\x0c\x44\x65viceUpdate\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"P\n\x12\x44\x65viceDisconnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xd8\x01\n\x1ePhysicalPipelineReconciliation\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12Y\n\x14physical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x1f\n\x0b\x66orce_apply\x18\x03 \x01(\x08R\nforceApply\"\xbc\x01\n PipelineDeploymentReconciliation\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x37\n\x08\x66leet_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\"U\n\x17\x42\x61seConfigUpdateRequest\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\
|
24
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/api/corepb/v1/controlplane.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\xc4\x01\n\nCPMetadata\x12\x37\n\x05topic\x18\x01 \x01(\x0e\x32!.gml.internal.api.core.v1.CPTopicR\x05topic\x12:\n\tentity_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x45ntityIDR\x08\x65ntityId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"v\n\tCPMessage\x12@\n\x08metadata\x18\x01 \x01(\x0b\x32$.gml.internal.api.core.v1.CPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"M\n\x0f\x44\x65viceConnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"J\n\x0c\x44\x65viceUpdate\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"P\n\x12\x44\x65viceDisconnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xd8\x01\n\x1ePhysicalPipelineReconciliation\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12Y\n\x14physical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x1f\n\x0b\x66orce_apply\x18\x03 \x01(\x08R\nforceApply\"\xbc\x01\n PipelineDeploymentReconciliation\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x37\n\x08\x66leet_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\"U\n\x17\x42\x61seConfigUpdateRequest\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xb2\x03\n\x1bPipelineCompilationMetadata\x12\x8c\x01\n\x1aphysical_pipeline_metadata\x18\x01 \x01(\x0b\x32N.gml.internal.api.core.v1.PipelineCompilationMetadata.PhysicalPipelineMetadataR\x18physicalPipelineMetadata\x1a\x83\x02\n\x18PhysicalPipelineMetadata\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x03 \x01(\tR\x12\x64\x65viceResourceHash\x12_\n\x16pipeline_deployment_id\x18\x04 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\"\x88\x04\n\x1aPipelineCompilationRequest\x12Q\n\x08metadata\x18\x01 \x01(\x0b\x32\x35.gml.internal.api.core.v1.PipelineCompilationMetadataR\x08metadata\x12V\n\x13logical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x03 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12]\n\x13\x64\x65vice_capabilities\x18\x04 \x01(\x0b\x32,.gml.internal.api.core.v1.DeviceCapabilitiesR\x12\x64\x65viceCapabilities\x12\x37\n\x06models\x18\x05 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ModelR\x06models\x12Q\n\ngem_config\x18\x06 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigB\r\xe2\xde\x1f\tGEMConfigR\tgemConfig\"\xe1\x01\n\x1bPipelineCompilationResponse\x12Q\n\x08metadata\x18\x01 \x01(\x0b\x32\x35.gml.internal.api.core.v1.PipelineCompilationMetadataR\x08metadata\x12\x44\n\texec_spec\x18\x02 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x08\x65xecSpec\x12)\n\x06status\x18\x03 \x01(\x0b\x32\x11.gml.types.StatusR\x06status*\xcf\x02\n\x07\x43PTopic\x12\x14\n\x10\x43P_TOPIC_UNKNOWN\x10\x00\x12\x1d\n\x19\x43P_TOPIC_DEVICE_CONNECTED\x10\x01\x12-\n)CP_TOPIC_PHYSICAL_PIPELINE_RECONCILIATION\x10\x02\x12 \n\x1c\x43P_TOPIC_DEVICE_DISCONNECTED\x10\x03\x12/\n+CP_TOPIC_PIPELINE_DEPLOYMENT_RECONCILIATION\x10\x04\x12\x1a\n\x16\x43P_TOPIC_DEVICE_UPDATE\x10\x05\x12\x1a\n\x16\x43P_TOPIC_DEVICE_CONFIG\x10\x06\x12)\n%CP_TOPIC_PIPELINE_COMPILATION_REQUEST\x10\x07\x12*\n&CP_TOPIC_PIPELINE_COMPILATION_RESPONSE\x10\x08\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
24
25
|
|
25
26
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
26
27
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.controlplane_pb2', globals())
|
@@ -46,34 +47,38 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
46
47
|
_PIPELINEDEPLOYMENTRECONCILIATION.fields_by_name['fleet_id']._serialized_options = b'\342\336\037\007FleetID'
|
47
48
|
_BASECONFIGUPDATEREQUEST.fields_by_name['device_id']._options = None
|
48
49
|
_BASECONFIGUPDATEREQUEST.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
|
49
|
-
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['
|
50
|
-
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['
|
50
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['device_id']._options = None
|
51
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
|
52
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['pipeline_deployment_id']._options = None
|
53
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA.fields_by_name['pipeline_deployment_id']._serialized_options = b'\342\336\037\024PipelineDeploymentID'
|
51
54
|
_PIPELINECOMPILATIONREQUEST.fields_by_name['logical_pipeline_id']._options = None
|
52
55
|
_PIPELINECOMPILATIONREQUEST.fields_by_name['logical_pipeline_id']._serialized_options = b'\342\336\037\021LogicalPipelineID'
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
|
66
|
-
|
67
|
-
|
68
|
-
|
69
|
-
|
70
|
-
|
71
|
-
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
76
|
-
|
77
|
-
|
78
|
-
|
56
|
+
_PIPELINECOMPILATIONREQUEST.fields_by_name['gem_config']._options = None
|
57
|
+
_PIPELINECOMPILATIONREQUEST.fields_by_name['gem_config']._serialized_options = b'\342\336\037\tGEMConfig'
|
58
|
+
_CPTOPIC._serialized_start=2559
|
59
|
+
_CPTOPIC._serialized_end=2894
|
60
|
+
_CPMETADATA._serialized_start=318
|
61
|
+
_CPMETADATA._serialized_end=514
|
62
|
+
_CPMESSAGE._serialized_start=516
|
63
|
+
_CPMESSAGE._serialized_end=634
|
64
|
+
_DEVICECONNECTED._serialized_start=636
|
65
|
+
_DEVICECONNECTED._serialized_end=713
|
66
|
+
_DEVICEUPDATE._serialized_start=715
|
67
|
+
_DEVICEUPDATE._serialized_end=789
|
68
|
+
_DEVICEDISCONNECTED._serialized_start=791
|
69
|
+
_DEVICEDISCONNECTED._serialized_end=871
|
70
|
+
_PHYSICALPIPELINERECONCILIATION._serialized_start=874
|
71
|
+
_PHYSICALPIPELINERECONCILIATION._serialized_end=1090
|
72
|
+
_PIPELINEDEPLOYMENTRECONCILIATION._serialized_start=1093
|
73
|
+
_PIPELINEDEPLOYMENTRECONCILIATION._serialized_end=1281
|
74
|
+
_BASECONFIGUPDATEREQUEST._serialized_start=1283
|
75
|
+
_BASECONFIGUPDATEREQUEST._serialized_end=1368
|
76
|
+
_PIPELINECOMPILATIONMETADATA._serialized_start=1371
|
77
|
+
_PIPELINECOMPILATIONMETADATA._serialized_end=1805
|
78
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA._serialized_start=1546
|
79
|
+
_PIPELINECOMPILATIONMETADATA_PHYSICALPIPELINEMETADATA._serialized_end=1805
|
80
|
+
_PIPELINECOMPILATIONREQUEST._serialized_start=1808
|
81
|
+
_PIPELINECOMPILATIONREQUEST._serialized_end=2328
|
82
|
+
_PIPELINECOMPILATIONRESPONSE._serialized_start=2331
|
83
|
+
_PIPELINECOMPILATIONRESPONSE._serialized_end=2556
|
79
84
|
# @@protoc_insertion_point(module_scope)
|
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
|
|
14
14
|
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
15
15
|
|
16
16
|
|
17
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/api/corepb/v1/device_info.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\"R\n\x10\x43\x61meraDriverInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\"\xb4\x02\n\nCameraMode\x12\x16\n\x06\x66ourcc\x18\x01 \x01(\tR\x06\x66ourcc\x12Q\n\x0bresolutions\x18\x02 \x03(\x0b\x32/.gml.internal.api.core.v1.CameraMode.ResolutionR\x0bresolutions\x1a\xba\x01\n\nResolution\x12\x14\n\x05width\x18\x01 \x01(\x05R\x05width\x12\x16\n\x06height\x18\x02 \x01(\x05R\x06height\x12V\n\tintervals\x18\x03 \x03(\x0b\x32\x38.gml.internal.api.core.v1.CameraMode.Resolution.IntervalR\tintervals\x1a&\n\x08Interval\x12\x1a\n\x08interval\x18\x01 \x01(\x01R\x08interval\"\xb3\x01\n\nCameraInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\x12)\n\tcamera_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08\x43\x61meraIDR\x08\x63\x61meraId\x12:\n\x05modes\x18\x03 \x03(\x0b\x32$.gml.internal.api.core.v1.CameraModeR\x05modes\"d\n\x15IREELLVMCPUTargetInfo\x12\x16\n\x06triple\x18\x01 \x01(\tR\x06triple\x12\x10\n\x03\x63pu\x18\x02 \x01(\tR\x03\x63pu\x12!\n\x0c\x63pu_features\x18\x03 \x03(\tR\x0b\x63puFeatures\"(\n\x12IREEROCmTargetInfo\x12\x12\n\x04\x61rch\x18\x01 \x01(\tR\x04\x61rch\"\x9f\x02\n\x0fIREERuntimeInfo\x12\x1b\n\tdevice_id\x18\x01 \x01(\x04R\x08\x64\x65viceId\x12<\n\x06target\x18\x02 \x01(\x0e\x32$.gml.internal.api.core.v1.IREETargetR\x06target\x12Y\n\x10llvm_target_info\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.IREELLVMCPUTargetInfoR\x0ellvmTargetInfo\x12V\n\x10rocm_target_info\x18\x04 \x01(\x0b\x32,.gml.internal.api.core.v1.IREEROCmTargetInfoR\x0erocmTargetInfo\"\
|
17
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/api/corepb/v1/device_info.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\"R\n\x10\x43\x61meraDriverInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\"\xb4\x02\n\nCameraMode\x12\x16\n\x06\x66ourcc\x18\x01 \x01(\tR\x06\x66ourcc\x12Q\n\x0bresolutions\x18\x02 \x03(\x0b\x32/.gml.internal.api.core.v1.CameraMode.ResolutionR\x0bresolutions\x1a\xba\x01\n\nResolution\x12\x14\n\x05width\x18\x01 \x01(\x05R\x05width\x12\x16\n\x06height\x18\x02 \x01(\x05R\x06height\x12V\n\tintervals\x18\x03 \x03(\x0b\x32\x38.gml.internal.api.core.v1.CameraMode.Resolution.IntervalR\tintervals\x1a&\n\x08Interval\x12\x1a\n\x08interval\x18\x01 \x01(\x01R\x08interval\"\xb3\x01\n\nCameraInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\x12)\n\tcamera_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08\x43\x61meraIDR\x08\x63\x61meraId\x12:\n\x05modes\x18\x03 \x03(\x0b\x32$.gml.internal.api.core.v1.CameraModeR\x05modes\"d\n\x15IREELLVMCPUTargetInfo\x12\x16\n\x06triple\x18\x01 \x01(\tR\x06triple\x12\x10\n\x03\x63pu\x18\x02 \x01(\tR\x03\x63pu\x12!\n\x0c\x63pu_features\x18\x03 \x03(\tR\x0b\x63puFeatures\"(\n\x12IREEROCmTargetInfo\x12\x12\n\x04\x61rch\x18\x01 \x01(\tR\x04\x61rch\"\x9f\x02\n\x0fIREERuntimeInfo\x12\x1b\n\tdevice_id\x18\x01 \x01(\x04R\x08\x64\x65viceId\x12<\n\x06target\x18\x02 \x01(\x0e\x32$.gml.internal.api.core.v1.IREETargetR\x06target\x12Y\n\x10llvm_target_info\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.IREELLVMCPUTargetInfoR\x0ellvmTargetInfo\x12V\n\x10rocm_target_info\x18\x04 \x01(\x0b\x32,.gml.internal.api.core.v1.IREEROCmTargetInfoR\x0erocmTargetInfo\"t\n\x13OpenVINORuntimeInfo\x12\x1b\n\tdevice_id\x18\x01 \x01(\tR\x08\x64\x65viceId\x12@\n\x06target\x18\x02 \x01(\x0e\x32(.gml.internal.api.core.v1.OpenVINOTargetR\x06target\"\xee\x01\n\x10ModelRuntimeInfo\x12>\n\x04type\x18\x01 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x04type\x12\x46\n\tiree_info\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.IREERuntimeInfoR\x08ireeInfo\x12R\n\ropenvino_info\x18\x03 \x01(\x0b\x32-.gml.internal.api.core.v1.OpenVINORuntimeInfoR\x0copenvinoInfo*Z\n\x0c\x43\x61meraDriver\x12\x19\n\x15\x43\x41MERA_DRIVER_UNKNOWN\x10\x00\x12\x17\n\x13\x43\x41MERA_DRIVER_ARGUS\x10\x01\x12\x16\n\x12\x43\x41MERA_DRIVER_V4L2\x10\x02*\xdf\x01\n\x10ModelRuntimeType\x12\x1e\n\x1aMODEL_RUNTIME_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_TENSORRT\x10\x01\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_OPENVINO\x10\x02\x12\x1b\n\x17MODEL_RUNTIME_TYPE_IREE\x10\x03\x12,\n(MODEL_RUNTIME_TYPE_HUGGINGFACE_TOKENIZER\x10\x04\x12\x1e\n\x1aMODEL_RUNTIME_TYPE_HAILORT\x10\x05*k\n\nIREETarget\x12\x17\n\x13IREE_TARGET_UNKNOWN\x10\x00\x12\x18\n\x14IREE_TARGET_LLVM_CPU\x10\x01\x12\x14\n\x10IREE_TARGET_ROCM\x10\x02\x12\x14\n\x10IREE_TARGET_VMVX\x10\x03*b\n\x0eOpenVINOTarget\x12\x1c\n\x18OPEN_VINO_TARGET_UNKNOWN\x10\x00\x12\x18\n\x14OPEN_VINO_TARGET_CPU\x10\x01\x12\x18\n\x14OPEN_VINO_TARGET_GPU\x10\x02\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
18
18
|
|
19
19
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
20
20
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.device_info_pb2', globals())
|
@@ -24,12 +24,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
24
24
|
DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
|
25
25
|
_CAMERAINFO.fields_by_name['camera_id']._options = None
|
26
26
|
_CAMERAINFO.fields_by_name['camera_id']._serialized_options = b'\342\336\037\010CameraID'
|
27
|
-
_CAMERADRIVER._serialized_start=
|
28
|
-
_CAMERADRIVER._serialized_end=
|
29
|
-
_MODELRUNTIMETYPE._serialized_start=
|
30
|
-
_MODELRUNTIMETYPE._serialized_end=
|
31
|
-
_IREETARGET._serialized_start=
|
32
|
-
_IREETARGET._serialized_end=
|
27
|
+
_CAMERADRIVER._serialized_start=1457
|
28
|
+
_CAMERADRIVER._serialized_end=1547
|
29
|
+
_MODELRUNTIMETYPE._serialized_start=1550
|
30
|
+
_MODELRUNTIMETYPE._serialized_end=1773
|
31
|
+
_IREETARGET._serialized_start=1775
|
32
|
+
_IREETARGET._serialized_end=1882
|
33
|
+
_OPENVINOTARGET._serialized_start=1884
|
34
|
+
_OPENVINOTARGET._serialized_end=1982
|
33
35
|
_CAMERADRIVERINFO._serialized_start=87
|
34
36
|
_CAMERADRIVERINFO._serialized_end=169
|
35
37
|
_CAMERAMODE._serialized_start=172
|
@@ -46,6 +48,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
46
48
|
_IREEROCMTARGETINFO._serialized_end=806
|
47
49
|
_IREERUNTIMEINFO._serialized_start=809
|
48
50
|
_IREERUNTIMEINFO._serialized_end=1096
|
49
|
-
|
50
|
-
|
51
|
+
_OPENVINORUNTIMEINFO._serialized_start=1098
|
52
|
+
_OPENVINORUNTIMEINFO._serialized_end=1214
|
53
|
+
_MODELRUNTIMEINFO._serialized_start=1217
|
54
|
+
_MODELRUNTIMEINFO._serialized_end=1455
|
51
55
|
# @@protoc_insertion_point(module_scope)
|
@@ -12,9 +12,10 @@ _sym_db = _symbol_database.Default()
|
|
12
12
|
|
13
13
|
|
14
14
|
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
15
|
+
from gml.proto.src.api.corepb.v1 import device_info_pb2 as src_dot_api_dot_corepb_dot_v1_dot_device__info__pb2
|
15
16
|
|
16
17
|
|
17
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/gem_config.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\"w\n\x0bVideoSource\x12=\n\x04type\x18\x01 \x01(\x0e\x32).gml.internal.api.core.v1.VideoSourceTypeR\x04type\x12)\n\tsource_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08SourceIDR\x08sourceId\"\
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/gem_config.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a#src/api/corepb/v1/device_info.proto\"w\n\x0bVideoSource\x12=\n\x04type\x18\x01 \x01(\x0e\x32).gml.internal.api.core.v1.VideoSourceTypeR\x04type\x12)\n\tsource_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08SourceIDR\x08sourceId\"\xbd\x02\n\x12ModelRuntimeConfig\x12>\n\x04type\x18\x01 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x04type\x12U\n\x0biree_target\x18\x02 \x01(\x0e\x32$.gml.internal.api.core.v1.IREETargetB\x0e\xe2\xde\x1f\nIREETargetR\nireeTarget\x12\x65\n\x0fopenvino_target\x18\x03 \x01(\x0e\x32(.gml.internal.api.core.v1.OpenVINOTargetB\x12\xe2\xde\x1f\x0eOpenVINOTargetR\x0eopenvinoTarget\x12)\n\tdevice_id\x18\x04 \x01(\tB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xb9\x02\n\tGEMConfig\x12j\n\x13video_source_claims\x18\x01 \x03(\x0b\x32:.gml.internal.api.core.v1.GEMConfig.VideoSourceClaimsEntryR\x11videoSourceClaims\x12S\n\x0emodel_runtimes\x18\x02 \x03(\x0b\x32,.gml.internal.api.core.v1.ModelRuntimeConfigR\rmodelRuntimes\x1ak\n\x16VideoSourceClaimsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12;\n\x05value\x18\x02 \x01(\x0b\x32%.gml.internal.api.core.v1.VideoSourceR\x05value:\x02\x38\x01\"$\n\x0c\x43onfigStatus\x12\x14\n\x05\x65rror\x18\x01 \x01(\tR\x05\x65rror\"\x95\x02\n\x0eGEMConfigState\x12J\n\x0e\x61pplied_config\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigR\rappliedConfig\x12R\n\x08statuses\x18\x02 \x03(\x0b\x32\x36.gml.internal.api.core.v1.GEMConfigState.StatusesEntryR\x08statuses\x1a\x63\n\rStatusesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12<\n\x05value\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.ConfigStatusR\x05value:\x02\x38\x01*\xa4\x01\n\x0fVideoSourceType\x12\x1d\n\x19VIDEO_SOURCE_TYPE_UNKNOWN\x10\x00\x12\x1b\n\x17VIDEO_SOURCE_TYPE_ARGUS\x10\x01\x12\x1a\n\x16VIDEO_SOURCE_TYPE_V4L2\x10\x02\x12\x1a\n\x16VIDEO_SOURCE_TYPE_FILE\x10\x03\x12\x1d\n\x19VIDEO_SOURCE_TYPE_NETWORK\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
18
19
|
|
19
20
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
20
21
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.gem_config_pb2', globals())
|
@@ -24,22 +25,30 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
24
25
|
DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
|
25
26
|
_VIDEOSOURCE.fields_by_name['source_id']._options = None
|
26
27
|
_VIDEOSOURCE.fields_by_name['source_id']._serialized_options = b'\342\336\037\010SourceID'
|
28
|
+
_MODELRUNTIMECONFIG.fields_by_name['iree_target']._options = None
|
29
|
+
_MODELRUNTIMECONFIG.fields_by_name['iree_target']._serialized_options = b'\342\336\037\nIREETarget'
|
30
|
+
_MODELRUNTIMECONFIG.fields_by_name['openvino_target']._options = None
|
31
|
+
_MODELRUNTIMECONFIG.fields_by_name['openvino_target']._serialized_options = b'\342\336\037\016OpenVINOTarget'
|
32
|
+
_MODELRUNTIMECONFIG.fields_by_name['device_id']._options = None
|
33
|
+
_MODELRUNTIMECONFIG.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
|
27
34
|
_GEMCONFIG_VIDEOSOURCECLAIMSENTRY._options = None
|
28
35
|
_GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_options = b'8\001'
|
29
36
|
_GEMCONFIGSTATE_STATUSESENTRY._options = None
|
30
37
|
_GEMCONFIGSTATE_STATUSESENTRY._serialized_options = b'8\001'
|
31
|
-
_VIDEOSOURCETYPE._serialized_start=
|
32
|
-
_VIDEOSOURCETYPE._serialized_end=
|
33
|
-
_VIDEOSOURCE._serialized_start=
|
34
|
-
_VIDEOSOURCE._serialized_end=
|
35
|
-
|
36
|
-
|
37
|
-
|
38
|
-
|
39
|
-
|
40
|
-
|
41
|
-
|
42
|
-
|
43
|
-
|
44
|
-
|
38
|
+
_VIDEOSOURCETYPE._serialized_start=1199
|
39
|
+
_VIDEOSOURCETYPE._serialized_end=1363
|
40
|
+
_VIDEOSOURCE._serialized_start=123
|
41
|
+
_VIDEOSOURCE._serialized_end=242
|
42
|
+
_MODELRUNTIMECONFIG._serialized_start=245
|
43
|
+
_MODELRUNTIMECONFIG._serialized_end=562
|
44
|
+
_GEMCONFIG._serialized_start=565
|
45
|
+
_GEMCONFIG._serialized_end=878
|
46
|
+
_GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_start=771
|
47
|
+
_GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_end=878
|
48
|
+
_CONFIGSTATUS._serialized_start=880
|
49
|
+
_CONFIGSTATUS._serialized_end=916
|
50
|
+
_GEMCONFIGSTATE._serialized_start=919
|
51
|
+
_GEMCONFIGSTATE._serialized_end=1196
|
52
|
+
_GEMCONFIGSTATE_STATUSESENTRY._serialized_start=1097
|
53
|
+
_GEMCONFIGSTATE_STATUSESENTRY._serialized_end=1196
|
45
54
|
# @@protoc_insertion_point(module_scope)
|
@@ -23,7 +23,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
|
|
23
23
|
from gml.proto.src.api.corepb.v1 import device_info_pb2 as src_dot_api_dot_corepb_dot_v1_dot_device__info__pb2
|
24
24
|
|
25
25
|
|
26
|
-
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\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a#src/api/corepb/v1/device_info.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\"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\"\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\"\xcd\x02\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\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12M\n\x0c\x65xec_runtime\x18\x06 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12\x1c\n\x07runtime\x18\x04 \x01(\tB\x02\x18\x01R\x07runtime\"\xaa\x02\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\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12O\n\rexec_runtimes\x18\x06 \x03(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0c\x65xecRuntimes\x12\x1c\n\x07runtime\x18\x03 \x01(\tB\x02\x18\x01R\x07runtime\"\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\"\xfc\x04\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\x12M\n\x0c\x65xec_runtime\x18\x33 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\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\x12W\n\tiree_spec\x18\x66 \x01(\x0b\x32\'.gml.internal.api.core.v1.IREEModelSpecB\x11\xe2\xde\x1f\rIREEModelSpecR\x08ireeSpec\x12\x1c\n\x07runtime\x18\x32 \x01(\tB\x02\x18\x01R\x07runtime\"\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\"M\n\rIREEModelSpec\x12\x1f\n\x0b\x64river_name\x18\x01 \x01(\tR\ndriverName\x12\x1b\n\tdevice_id\x18\x02 \x01(\x04R\x08\x64\x65viceId*\x9c\x08\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_ESTIMATE_DEPTH\x10\xf3\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
|
+
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\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a#src/api/corepb/v1/device_info.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\"\xba\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\x99\x02\n\x12ImageChannelParams\x12j\n\x06\x66ormat\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParams.ImageChannelFormatR\x06\x66ormat\"\x96\x01\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\x12\"\n\x1eIMAGE_CHANNEL_FORMAT_GRAYSCALE\x10\x03\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\"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\"\xfd\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\"\xd8\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\x12\x16\n\x12MODEL_KIND_HAILORT\x10\x07\"\xd1\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\x12\x1c\n\x18MODEL_STORAGE_FORMAT_HEF\x10\x08\"\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\"\xcd\x02\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\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12M\n\x0c\x65xec_runtime\x18\x06 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12\x1c\n\x07runtime\x18\x04 \x01(\tB\x02\x18\x01R\x07runtime\"\x88\x04\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\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12\x66\n\rruntime_specs\x18\x07 \x03(\x0b\x32\x41.gml.internal.api.core.v1.PhysicalPipelineStatus.ModelRuntimeSpecR\x0cruntimeSpecs\x12\x1c\n\x07runtime\x18\x03 \x01(\tB\x02\x18\x01R\x07runtime\x12S\n\rexec_runtimes\x18\x06 \x03(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeB\x02\x18\x01R\x0c\x65xecRuntimes\x1ap\n\x10ModelRuntimeSpec\x12\x44\n\x07runtime\x18\x01 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x07runtime\x12\x16\n\x06\x64\x65vice\x18\x02 \x01(\tR\x06\x64\x65vice\"\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\"\xfc\x04\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\x12M\n\x0c\x65xec_runtime\x18\x33 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\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\x12W\n\tiree_spec\x18\x66 \x01(\x0b\x32\'.gml.internal.api.core.v1.IREEModelSpecB\x11\xe2\xde\x1f\rIREEModelSpecR\x08ireeSpec\x12\x1c\n\x07runtime\x18\x32 \x01(\tB\x02\x18\x01R\x07runtime\"\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\"\xab\x01\n\x11OpenVINOModelSpec\x12X\n\x0binput_shape\x18\x01 \x03(\x0b\x32\x37.gml.internal.api.core.v1.OpenVINOModelSpec.TensorShapeR\ninputShape\x12\x1b\n\tdevice_id\x18\x02 \x01(\tR\x08\x64\x65viceId\x1a\x1f\n\x0bTensorShape\x12\x10\n\x03\x64im\x18\x01 \x03(\x05R\x03\x64im\"M\n\rIREEModelSpec\x12\x1f\n\x0b\x64river_name\x18\x01 \x01(\tR\ndriverName\x12\x1b\n\tdevice_id\x18\x02 \x01(\x04R\x08\x64\x65viceId*\xd2\x08\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_ESTIMATE_DEPTH\x10\xf3\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\x34\n/LOGICAL_PIPELINE_NODE_KIND_FILTER_MOVING_TRACKS\x10\xe1\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')
|
27
27
|
|
28
28
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
29
29
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.model_exec_pb2', globals())
|
@@ -55,6 +55,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
55
55
|
_PHYSICALPIPELINESPEC.fields_by_name['runtime']._serialized_options = b'\030\001'
|
56
56
|
_PHYSICALPIPELINESTATUS.fields_by_name['runtime']._options = None
|
57
57
|
_PHYSICALPIPELINESTATUS.fields_by_name['runtime']._serialized_options = b'\030\001'
|
58
|
+
_PHYSICALPIPELINESTATUS.fields_by_name['exec_runtimes']._options = None
|
59
|
+
_PHYSICALPIPELINESTATUS.fields_by_name['exec_runtimes']._serialized_options = b'\030\001'
|
58
60
|
_FILERESOURCE.fields_by_name['file_id']._options = None
|
59
61
|
_FILERESOURCE.fields_by_name['file_id']._serialized_options = b'\342\336\037\006FileID'
|
60
62
|
_MODELSPEC.fields_by_name['onnx_blob_key']._options = None
|
@@ -69,10 +71,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
69
71
|
_MODELSPEC.fields_by_name['iree_spec']._serialized_options = b'\342\336\037\rIREEModelSpec'
|
70
72
|
_MODELSPEC.fields_by_name['runtime']._options = None
|
71
73
|
_MODELSPEC.fields_by_name['runtime']._serialized_options = b'\030\001'
|
72
|
-
_LOGICALPIPELINENODEKIND._serialized_start=
|
73
|
-
_LOGICALPIPELINENODEKIND._serialized_end=
|
74
|
-
_PIPELINESTATE._serialized_start=
|
75
|
-
_PIPELINESTATE._serialized_end=
|
74
|
+
_LOGICALPIPELINENODEKIND._serialized_start=11924
|
75
|
+
_LOGICALPIPELINENODEKIND._serialized_end=13030
|
76
|
+
_PIPELINESTATE._serialized_start=13033
|
77
|
+
_PIPELINESTATE._serialized_end=13248
|
76
78
|
_BOUNDINGBOXINFO._serialized_start=258
|
77
79
|
_BOUNDINGBOXINFO._serialized_end=553
|
78
80
|
_BOUNDINGBOXINFO_BOUNDINGBOXFORMAT._serialized_start=409
|
@@ -90,97 +92,99 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
90
92
|
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_start=1444
|
91
93
|
_IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_end=1633
|
92
94
|
_DIMENSIONSEMANTICS._serialized_start=1636
|
93
|
-
_DIMENSIONSEMANTICS._serialized_end=
|
95
|
+
_DIMENSIONSEMANTICS._serialized_end=4382
|
94
96
|
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_start=2359
|
95
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_end=
|
96
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_start=
|
97
|
-
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_end=
|
98
|
-
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_start=
|
99
|
-
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_end=
|
100
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_start=
|
101
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_end=
|
102
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_start=
|
103
|
-
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_end=
|
104
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_start=
|
105
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_end=
|
106
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_start=
|
107
|
-
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_end=
|
108
|
-
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_start=
|
109
|
-
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_end=
|
110
|
-
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_start=
|
111
|
-
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=
|
112
|
-
_TENSORSEMANTICS._serialized_start=
|
113
|
-
_TENSORSEMANTICS._serialized_end=
|
114
|
-
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_start=
|
115
|
-
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_end=
|
116
|
-
_GENERATIONCONFIG._serialized_start=
|
117
|
-
_GENERATIONCONFIG._serialized_end=
|
118
|
-
_MODEL._serialized_start=
|
119
|
-
_MODEL._serialized_end=
|
120
|
-
_MODELINFO._serialized_start=
|
121
|
-
_MODELINFO._serialized_end=
|
122
|
-
_MODELINFO_FILEASSETSENTRY._serialized_start=
|
123
|
-
_MODELINFO_FILEASSETSENTRY._serialized_end=
|
124
|
-
_MODELINFO_MODELKIND._serialized_start=
|
125
|
-
_MODELINFO_MODELKIND._serialized_end=
|
126
|
-
_MODELINFO_MODELSTORAGEFORMAT._serialized_start=
|
127
|
-
_MODELINFO_MODELSTORAGEFORMAT._serialized_end=
|
128
|
-
_NODE._serialized_start=
|
129
|
-
_NODE._serialized_end=
|
130
|
-
_PARAMREF._serialized_start=
|
131
|
-
_PARAMREF._serialized_end=
|
132
|
-
_VALUE._serialized_start=
|
133
|
-
_VALUE._serialized_end=
|
134
|
-
_VALUE_MODELREF._serialized_start=
|
135
|
-
_VALUE_MODELREF._serialized_end=
|
136
|
-
_VALUE_LAMBDA._serialized_start=
|
137
|
-
_VALUE_LAMBDA._serialized_end=
|
138
|
-
_NODEATTRIBUTES._serialized_start=
|
139
|
-
_NODEATTRIBUTES._serialized_end=
|
140
|
-
_NODEINPUT._serialized_start=
|
141
|
-
_NODEINPUT._serialized_end=
|
142
|
-
_NODEINPUT_NODEOUTPUTREF._serialized_start=
|
143
|
-
_NODEINPUT_NODEOUTPUTREF._serialized_end=
|
144
|
-
_NODEINPUT_LAMBDAINPUTREF._serialized_start=
|
145
|
-
_NODEINPUT_LAMBDAINPUTREF._serialized_end=
|
146
|
-
_NODEOUTPUT._serialized_start=
|
147
|
-
_NODEOUTPUT._serialized_end=
|
148
|
-
_PARAM._serialized_start=
|
149
|
-
_PARAM._serialized_end=
|
150
|
-
_LOGICALPIPELINE._serialized_start=
|
151
|
-
_LOGICALPIPELINE._serialized_end=
|
152
|
-
_PIPELINEDEPLOYMENT._serialized_start=
|
153
|
-
_PIPELINEDEPLOYMENT._serialized_end=
|
154
|
-
_PIPELINEDEPLOYMENTSPEC._serialized_start=
|
155
|
-
_PIPELINEDEPLOYMENTSPEC._serialized_end=
|
156
|
-
_PIPELINEDEPLOYMENTSTATUS._serialized_start=
|
157
|
-
_PIPELINEDEPLOYMENTSTATUS._serialized_end=
|
158
|
-
_PHYSICALPIPELINE._serialized_start=
|
159
|
-
_PHYSICALPIPELINE._serialized_end=
|
160
|
-
_PHYSICALPIPELINESPEC._serialized_start=
|
161
|
-
_PHYSICALPIPELINESPEC._serialized_end=
|
162
|
-
_PHYSICALPIPELINESTATUS._serialized_start=
|
163
|
-
_PHYSICALPIPELINESTATUS._serialized_end=
|
164
|
-
|
165
|
-
|
166
|
-
|
167
|
-
|
168
|
-
|
169
|
-
|
170
|
-
|
171
|
-
|
172
|
-
|
173
|
-
|
174
|
-
|
175
|
-
|
176
|
-
|
177
|
-
|
178
|
-
|
179
|
-
|
180
|
-
|
181
|
-
|
182
|
-
|
183
|
-
|
184
|
-
|
185
|
-
|
97
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_end=2640
|
98
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_start=2490
|
99
|
+
_DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_end=2640
|
100
|
+
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_start=2642
|
101
|
+
_DIMENSIONSEMANTICS_DETECTIONCANDIDATESPARAMS._serialized_end=2703
|
102
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_start=2706
|
103
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS._serialized_end=3229
|
104
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_start=3175
|
105
|
+
_DIMENSIONSEMANTICS_DETECTIONOUTPUTPARAMS_INDEXRANGE._serialized_end=3229
|
106
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_start=3232
|
107
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS._serialized_end=3569
|
108
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_start=3369
|
109
|
+
_DIMENSIONSEMANTICS_SEGMENTATIONMASKPARAMS_SEGMENTATIONMASKKIND._serialized_end=3569
|
110
|
+
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_start=3571
|
111
|
+
_DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_end=3663
|
112
|
+
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_start=3666
|
113
|
+
_DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=4382
|
114
|
+
_TENSORSEMANTICS._serialized_start=4385
|
115
|
+
_TENSORSEMANTICS._serialized_end=4746
|
116
|
+
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_start=4566
|
117
|
+
_TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_end=4746
|
118
|
+
_GENERATIONCONFIG._serialized_start=4748
|
119
|
+
_GENERATIONCONFIG._serialized_end=4802
|
120
|
+
_MODEL._serialized_start=4804
|
121
|
+
_MODEL._serialized_end=4909
|
122
|
+
_MODELINFO._serialized_start=4912
|
123
|
+
_MODELINFO._serialized_end=6317
|
124
|
+
_MODELINFO_FILEASSETSENTRY._serialized_start=5680
|
125
|
+
_MODELINFO_FILEASSETSENTRY._serialized_end=5758
|
126
|
+
_MODELINFO_MODELKIND._serialized_start=5761
|
127
|
+
_MODELINFO_MODELKIND._serialized_end=5977
|
128
|
+
_MODELINFO_MODELSTORAGEFORMAT._serialized_start=5980
|
129
|
+
_MODELINFO_MODELSTORAGEFORMAT._serialized_end=6317
|
130
|
+
_NODE._serialized_start=6320
|
131
|
+
_NODE._serialized_end=6616
|
132
|
+
_PARAMREF._serialized_start=6618
|
133
|
+
_PARAMREF._serialized_end=6648
|
134
|
+
_VALUE._serialized_start=6651
|
135
|
+
_VALUE._serialized_end=7206
|
136
|
+
_VALUE_MODELREF._serialized_start=7013
|
137
|
+
_VALUE_MODELREF._serialized_end=7084
|
138
|
+
_VALUE_LAMBDA._serialized_start=7086
|
139
|
+
_VALUE_LAMBDA._serialized_end=7198
|
140
|
+
_NODEATTRIBUTES._serialized_start=7208
|
141
|
+
_NODEATTRIBUTES._serialized_end=7299
|
142
|
+
_NODEINPUT._serialized_start=7302
|
143
|
+
_NODEINPUT._serialized_end=7714
|
144
|
+
_NODEINPUT_NODEOUTPUTREF._serialized_start=7603
|
145
|
+
_NODEINPUT_NODEOUTPUTREF._serialized_end=7667
|
146
|
+
_NODEINPUT_LAMBDAINPUTREF._serialized_start=7669
|
147
|
+
_NODEINPUT_LAMBDAINPUTREF._serialized_end=7705
|
148
|
+
_NODEOUTPUT._serialized_start=7716
|
149
|
+
_NODEOUTPUT._serialized_end=7748
|
150
|
+
_PARAM._serialized_start=7750
|
151
|
+
_PARAM._serialized_end=7847
|
152
|
+
_LOGICALPIPELINE._serialized_start=7850
|
153
|
+
_LOGICALPIPELINE._serialized_end=8038
|
154
|
+
_PIPELINEDEPLOYMENT._serialized_start=8041
|
155
|
+
_PIPELINEDEPLOYMENT._serialized_end=8596
|
156
|
+
_PIPELINEDEPLOYMENTSPEC._serialized_start=8598
|
157
|
+
_PIPELINEDEPLOYMENTSPEC._serialized_end=8685
|
158
|
+
_PIPELINEDEPLOYMENTSTATUS._serialized_start=8687
|
159
|
+
_PIPELINEDEPLOYMENTSTATUS._serialized_end=8800
|
160
|
+
_PHYSICALPIPELINE._serialized_start=8803
|
161
|
+
_PHYSICALPIPELINE._serialized_end=9279
|
162
|
+
_PHYSICALPIPELINESPEC._serialized_start=9282
|
163
|
+
_PHYSICALPIPELINESPEC._serialized_end=9615
|
164
|
+
_PHYSICALPIPELINESTATUS._serialized_start=9618
|
165
|
+
_PHYSICALPIPELINESTATUS._serialized_end=10138
|
166
|
+
_PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_start=10026
|
167
|
+
_PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_end=10138
|
168
|
+
_FILERESOURCE._serialized_start=10141
|
169
|
+
_FILERESOURCE._serialized_end=10273
|
170
|
+
_EXECUTIONSPEC._serialized_start=10276
|
171
|
+
_EXECUTIONSPEC._serialized_end=10415
|
172
|
+
_NAMEDASSET._serialized_start=10417
|
173
|
+
_NAMEDASSET._serialized_end=10509
|
174
|
+
_MODELSPEC._serialized_start=10512
|
175
|
+
_MODELSPEC._serialized_end=11148
|
176
|
+
_TENSORRTMODELSPEC._serialized_start=11151
|
177
|
+
_TENSORRTMODELSPEC._serialized_end=11405
|
178
|
+
_TENSORRTOPTIMIZATIONPROFILE._serialized_start=11407
|
179
|
+
_TENSORRTOPTIMIZATIONPROFILE._serialized_end=11534
|
180
|
+
_TENSORRTTENSORSHAPERANGE._serialized_start=11536
|
181
|
+
_TENSORRTTENSORSHAPERANGE._serialized_end=11613
|
182
|
+
_TENSORRTMEMPOOLLIMITS._serialized_start=11615
|
183
|
+
_TENSORRTMEMPOOLLIMITS._serialized_end=11668
|
184
|
+
_OPENVINOMODELSPEC._serialized_start=11671
|
185
|
+
_OPENVINOMODELSPEC._serialized_end=11842
|
186
|
+
_OPENVINOMODELSPEC_TENSORSHAPE._serialized_start=11811
|
187
|
+
_OPENVINOMODELSPEC_TENSORSHAPE._serialized_end=11842
|
188
|
+
_IREEMODELSPEC._serialized_start=11844
|
189
|
+
_IREEMODELSPEC._serialized_end=11921
|
186
190
|
# @@protoc_insertion_point(module_scope)
|
@@ -13,11 +13,12 @@ _sym_db = _symbol_database.Default()
|
|
13
13
|
|
14
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
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.api.corepb.v1 import gem_config_pb2 as src_dot_api_dot_corepb_dot_v1_dot_gem__config__pb2
|
16
17
|
from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_dot_uuid__pb2
|
17
18
|
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
18
19
|
|
19
20
|
|
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\"\
|
21
|
+
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\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"\xa9\x03\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\x37\n\x06models\x18\x04 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ModelR\x06models\x12Q\n\ngem_config\x18\x05 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigB\r\xe2\xde\x1f\tGEMConfigR\tgemConfig\"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
|
|
22
23
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
23
24
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.compiler.cpb.v1.cpb_pb2', globals())
|
@@ -27,10 +28,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
27
28
|
DESCRIPTOR._serialized_options = b'Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpb'
|
28
29
|
_COMPILEREQUEST.fields_by_name['logical_pipeline_id']._options = None
|
29
30
|
_COMPILEREQUEST.fields_by_name['logical_pipeline_id']._serialized_options = b'\342\336\037\021LogicalPipelineID'
|
30
|
-
_COMPILEREQUEST.
|
31
|
-
_COMPILEREQUEST.
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
31
|
+
_COMPILEREQUEST.fields_by_name['gem_config']._options = None
|
32
|
+
_COMPILEREQUEST.fields_by_name['gem_config']._serialized_options = b'\342\336\037\tGEMConfig'
|
33
|
+
_COMPILEREQUEST._serialized_start=244
|
34
|
+
_COMPILEREQUEST._serialized_end=669
|
35
|
+
_COMPILERESPONSE._serialized_start=671
|
36
|
+
_COMPILERESPONSE._serialized_end=758
|
37
|
+
_COMPILERSERVICE._serialized_start=761
|
38
|
+
_COMPILERSERVICE._serialized_end=900
|
36
39
|
# @@protoc_insertion_point(module_scope)
|
gml/tensor.py
CHANGED
@@ -77,6 +77,10 @@ def chan_format_str_to_proto(chan_format: str):
|
|
77
77
|
return (
|
78
78
|
modelexecpb.DimensionSemantics.ImageChannelParams.IMAGE_CHANNEL_FORMAT_BGR
|
79
79
|
)
|
80
|
+
case "gray":
|
81
|
+
return (
|
82
|
+
modelexecpb.DimensionSemantics.ImageChannelParams.IMAGE_CHANNEL_FORMAT_GRAYSCALE
|
83
|
+
)
|
80
84
|
case _:
|
81
85
|
raise ValueError("Invalid channel_format format: {}".format(chan_format))
|
82
86
|
|
File without changes
|