mplang-nightly 0.1.dev269__py3-none-any.whl → 0.1.dev270__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.
- mplang/__init__.py +391 -17
- mplang/{v2/backends → backends}/__init__.py +9 -7
- mplang/{v2/backends → backends}/bfv_impl.py +6 -6
- mplang/{v2/backends → backends}/crypto_impl.py +6 -6
- mplang/{v2/backends → backends}/field_impl.py +5 -5
- mplang/{v2/backends → backends}/func_impl.py +4 -4
- mplang/{v2/backends → backends}/phe_impl.py +3 -3
- mplang/{v2/backends → backends}/simp_design.md +1 -1
- mplang/{v2/backends → backends}/simp_driver/__init__.py +5 -5
- mplang/{v2/backends → backends}/simp_driver/http.py +8 -8
- mplang/{v2/backends → backends}/simp_driver/mem.py +9 -9
- mplang/{v2/backends → backends}/simp_driver/ops.py +4 -4
- mplang/{v2/backends → backends}/simp_driver/state.py +2 -2
- mplang/{v2/backends → backends}/simp_driver/values.py +2 -2
- mplang/{v2/backends → backends}/simp_worker/__init__.py +3 -3
- mplang/{v2/backends → backends}/simp_worker/http.py +10 -10
- mplang/{v2/backends → backends}/simp_worker/mem.py +1 -1
- mplang/{v2/backends → backends}/simp_worker/ops.py +5 -5
- mplang/{v2/backends → backends}/simp_worker/state.py +2 -4
- mplang/{v2/backends → backends}/spu_impl.py +8 -8
- mplang/{v2/backends → backends}/spu_state.py +4 -4
- mplang/{v2/backends → backends}/store_impl.py +3 -3
- mplang/{v2/backends → backends}/table_impl.py +8 -8
- mplang/{v2/backends → backends}/tee_impl.py +6 -6
- mplang/{v2/backends → backends}/tensor_impl.py +6 -6
- mplang/{v2/cli.py → cli.py} +9 -9
- mplang/{v2/cli_guide.md → cli_guide.md} +12 -12
- mplang/{v2/dialects → dialects}/__init__.py +5 -5
- mplang/{v2/dialects → dialects}/bfv.py +6 -6
- mplang/{v2/dialects → dialects}/crypto.py +5 -5
- mplang/{v2/dialects → dialects}/dtypes.py +2 -2
- mplang/{v2/dialects → dialects}/field.py +3 -3
- mplang/{v2/dialects → dialects}/func.py +2 -2
- mplang/{v2/dialects → dialects}/phe.py +6 -6
- mplang/{v2/dialects → dialects}/simp.py +6 -6
- mplang/{v2/dialects → dialects}/spu.py +7 -7
- mplang/{v2/dialects → dialects}/store.py +2 -2
- mplang/{v2/dialects → dialects}/table.py +3 -3
- mplang/{v2/dialects → dialects}/tee.py +6 -6
- mplang/{v2/dialects → dialects}/tensor.py +5 -5
- mplang/{v2/edsl → edsl}/__init__.py +3 -3
- mplang/{v2/edsl → edsl}/context.py +6 -6
- mplang/{v2/edsl → edsl}/graph.py +5 -5
- mplang/{v2/edsl → edsl}/jit.py +2 -2
- mplang/{v2/edsl → edsl}/object.py +1 -1
- mplang/{v2/edsl → edsl}/primitive.py +5 -5
- mplang/{v2/edsl → edsl}/printer.py +1 -1
- mplang/{v2/edsl → edsl}/serde.py +1 -1
- mplang/{v2/edsl → edsl}/tracer.py +7 -7
- mplang/{v2/edsl → edsl}/typing.py +1 -1
- mplang/{v2/kernels → kernels}/ldpc.cpp +13 -13
- mplang/{v2/kernels → kernels}/okvs.cpp +4 -4
- mplang/{v2/kernels → kernels}/okvs_opt.cpp +31 -31
- mplang/{v2/kernels → kernels}/py_kernels.py +1 -1
- mplang/{v2/libs → libs}/collective.py +5 -5
- mplang/{v2/libs → libs}/device/__init__.py +1 -1
- mplang/{v2/libs → libs}/device/api.py +12 -12
- mplang/{v2/libs → libs}/ml/__init__.py +1 -1
- mplang/{v2/libs → libs}/ml/sgb.py +4 -4
- mplang/{v2/libs → libs}/mpc/__init__.py +3 -3
- mplang/{v2/libs → libs}/mpc/_utils.py +2 -2
- mplang/{v2/libs → libs}/mpc/analytics/aggregation.py +1 -1
- mplang/{v2/libs → libs}/mpc/analytics/groupby.py +2 -2
- mplang/{v2/libs → libs}/mpc/analytics/permutation.py +3 -3
- mplang/{v2/libs → libs}/mpc/ot/base.py +3 -3
- mplang/{v2/libs → libs}/mpc/ot/extension.py +2 -2
- mplang/{v2/libs → libs}/mpc/ot/silent.py +4 -4
- mplang/{v2/libs → libs}/mpc/psi/cuckoo.py +3 -3
- mplang/{v2/libs → libs}/mpc/psi/okvs.py +1 -1
- mplang/{v2/libs → libs}/mpc/psi/okvs_gct.py +3 -3
- mplang/{v2/libs → libs}/mpc/psi/oprf.py +3 -3
- mplang/{v2/libs → libs}/mpc/psi/rr22.py +7 -7
- mplang/{v2/libs → libs}/mpc/psi/unbalanced.py +4 -4
- mplang/{v2/libs → libs}/mpc/vole/gilboa.py +3 -3
- mplang/{v2/libs → libs}/mpc/vole/ldpc.py +2 -2
- mplang/{v2/libs → libs}/mpc/vole/silver.py +6 -6
- mplang/{v2/runtime → runtime}/interpreter.py +11 -11
- mplang/{v2/runtime → runtime}/value.py +2 -2
- mplang/{v1/runtime → utils}/__init__.py +18 -15
- mplang/{v1/utils → utils}/func_utils.py +1 -1
- {mplang_nightly-0.1.dev269.dist-info → mplang_nightly-0.1.dev270.dist-info}/METADATA +2 -2
- mplang_nightly-0.1.dev270.dist-info/RECORD +102 -0
- mplang/v1/__init__.py +0 -157
- mplang/v1/_device.py +0 -602
- mplang/v1/analysis/__init__.py +0 -37
- mplang/v1/analysis/diagram.py +0 -567
- mplang/v1/core/__init__.py +0 -157
- mplang/v1/core/cluster.py +0 -343
- mplang/v1/core/comm.py +0 -281
- mplang/v1/core/context_mgr.py +0 -50
- mplang/v1/core/dtypes.py +0 -335
- mplang/v1/core/expr/__init__.py +0 -80
- mplang/v1/core/expr/ast.py +0 -542
- mplang/v1/core/expr/evaluator.py +0 -581
- mplang/v1/core/expr/printer.py +0 -285
- mplang/v1/core/expr/transformer.py +0 -141
- mplang/v1/core/expr/utils.py +0 -78
- mplang/v1/core/expr/visitor.py +0 -85
- mplang/v1/core/expr/walk.py +0 -387
- mplang/v1/core/interp.py +0 -160
- mplang/v1/core/mask.py +0 -325
- mplang/v1/core/mpir.py +0 -965
- mplang/v1/core/mpobject.py +0 -117
- mplang/v1/core/mptype.py +0 -407
- mplang/v1/core/pfunc.py +0 -130
- mplang/v1/core/primitive.py +0 -877
- mplang/v1/core/table.py +0 -218
- mplang/v1/core/tensor.py +0 -75
- mplang/v1/core/tracer.py +0 -383
- mplang/v1/host.py +0 -130
- mplang/v1/kernels/__init__.py +0 -41
- mplang/v1/kernels/base.py +0 -125
- mplang/v1/kernels/basic.py +0 -240
- mplang/v1/kernels/context.py +0 -369
- mplang/v1/kernels/crypto.py +0 -122
- mplang/v1/kernels/fhe.py +0 -858
- mplang/v1/kernels/mock_tee.py +0 -72
- mplang/v1/kernels/phe.py +0 -1864
- mplang/v1/kernels/spu.py +0 -341
- mplang/v1/kernels/sql_duckdb.py +0 -44
- mplang/v1/kernels/stablehlo.py +0 -90
- mplang/v1/kernels/value.py +0 -626
- mplang/v1/ops/__init__.py +0 -35
- mplang/v1/ops/base.py +0 -424
- mplang/v1/ops/basic.py +0 -294
- mplang/v1/ops/crypto.py +0 -262
- mplang/v1/ops/fhe.py +0 -272
- mplang/v1/ops/jax_cc.py +0 -147
- mplang/v1/ops/nnx_cc.py +0 -168
- mplang/v1/ops/phe.py +0 -216
- mplang/v1/ops/spu.py +0 -151
- mplang/v1/ops/sql_cc.py +0 -303
- mplang/v1/ops/tee.py +0 -36
- mplang/v1/protos/v1alpha1/mpir_pb2.py +0 -63
- mplang/v1/protos/v1alpha1/mpir_pb2.pyi +0 -557
- mplang/v1/protos/v1alpha1/value_pb2.py +0 -34
- mplang/v1/protos/v1alpha1/value_pb2.pyi +0 -169
- mplang/v1/runtime/channel.py +0 -230
- mplang/v1/runtime/cli.py +0 -451
- mplang/v1/runtime/client.py +0 -456
- mplang/v1/runtime/communicator.py +0 -131
- mplang/v1/runtime/data_providers.py +0 -303
- mplang/v1/runtime/driver.py +0 -324
- mplang/v1/runtime/exceptions.py +0 -27
- mplang/v1/runtime/http_api.md +0 -56
- mplang/v1/runtime/link_comm.py +0 -196
- mplang/v1/runtime/server.py +0 -501
- mplang/v1/runtime/session.py +0 -270
- mplang/v1/runtime/simulation.py +0 -324
- mplang/v1/simp/__init__.py +0 -13
- mplang/v1/simp/api.py +0 -353
- mplang/v1/simp/mpi.py +0 -131
- mplang/v1/simp/party.py +0 -225
- mplang/v1/simp/random.py +0 -120
- mplang/v1/simp/smpc.py +0 -238
- mplang/v1/utils/__init__.py +0 -13
- mplang/v1/utils/crypto.py +0 -32
- mplang/v1/utils/spu_utils.py +0 -130
- mplang/v1/utils/table_utils.py +0 -185
- mplang/v2/__init__.py +0 -424
- mplang_nightly-0.1.dev269.dist-info/RECORD +0 -180
- /mplang/{v2/backends → backends}/channel.py +0 -0
- /mplang/{v2/edsl → edsl}/README.md +0 -0
- /mplang/{v2/edsl → edsl}/registry.py +0 -0
- /mplang/{v2/kernels → kernels}/Makefile +0 -0
- /mplang/{v2/kernels → kernels}/__init__.py +0 -0
- /mplang/{v2/kernels → kernels}/gf128.cpp +0 -0
- /mplang/{v2/libs → libs}/device/cluster.py +0 -0
- /mplang/{v2/libs → libs}/mpc/analytics/__init__.py +0 -0
- /mplang/{v2/libs → libs}/mpc/analytics/groupby.md +0 -0
- /mplang/{v2/libs → libs}/mpc/common/constants.py +0 -0
- /mplang/{v2/libs → libs}/mpc/ot/__init__.py +0 -0
- /mplang/{v2/libs → libs}/mpc/psi/__init__.py +0 -0
- /mplang/{v2/libs → libs}/mpc/vole/__init__.py +0 -0
- /mplang/{v2/runtime → runtime}/__init__.py +0 -0
- /mplang/{v2/runtime → runtime}/dialect_state.py +0 -0
- /mplang/{v2/runtime → runtime}/object_store.py +0 -0
- {mplang_nightly-0.1.dev269.dist-info → mplang_nightly-0.1.dev270.dist-info}/WHEEL +0 -0
- {mplang_nightly-0.1.dev269.dist-info → mplang_nightly-0.1.dev270.dist-info}/entry_points.txt +0 -0
- {mplang_nightly-0.1.dev269.dist-info → mplang_nightly-0.1.dev270.dist-info}/licenses/LICENSE +0 -0
|
@@ -1,63 +0,0 @@
|
|
|
1
|
-
# -*- coding: utf-8 -*-
|
|
2
|
-
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
-
# source: mplang/protos/v1alpha1/mpir.proto
|
|
4
|
-
# Protobuf Python Version: 5.26.1
|
|
5
|
-
"""Generated protocol buffer code."""
|
|
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
|
-
from google.protobuf.internal import builder as _builder
|
|
10
|
-
# @@protoc_insertion_point(imports)
|
|
11
|
-
|
|
12
|
-
_sym_db = _symbol_database.Default()
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!mplang/protos/v1alpha1/mpir.proto\x12\x16mplang.protos.v1alpha1\"\xde\x03\n\tAttrProto\x12>\n\x04type\x18\x01 \x01(\x0e\x32*.mplang.protos.v1alpha1.AttrProto.AttrTypeR\x04type\x12\x0c\n\x01\x66\x18\x02 \x01(\x02R\x01\x66\x12\x0c\n\x01i\x18\x03 \x01(\x03R\x01i\x12\x0c\n\x01s\x18\x04 \x01(\tR\x01s\x12\x0c\n\x01\x62\x18\x05 \x01(\x08R\x01\x62\x12\x1b\n\traw_bytes\x18\x06 \x01(\x0cR\x08rawBytes\x12\x16\n\x06\x66loats\x18\x07 \x03(\x02R\x06\x66loats\x12\x12\n\x04ints\x18\x08 \x03(\x03R\x04ints\x12\x12\n\x04strs\x18\t \x03(\tR\x04strs\x12\x35\n\x04\x66unc\x18\n \x01(\x0b\x32!.mplang.protos.v1alpha1.FuncProtoR\x04\x66unc\x12\x38\n\x05graph\x18\x0b \x01(\x0b\x32\".mplang.protos.v1alpha1.GraphProtoR\x05graph\"\x8a\x01\n\x08\x41ttrType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\n\n\x06STRING\x10\x03\x12\x08\n\x04\x42OOL\x10\x04\x12\t\n\x05\x42YTES\x10\x05\x12\n\n\x06\x46LOATS\x10\x06\x12\x08\n\x04INTS\x10\x07\x12\x0b\n\x07STRINGS\x10\x08\x12\x0c\n\x08\x46UNCTION\x10\n\x12\t\n\x05GRAPH\x10\x0b\"\x87\x02\n\tFuncProto\x12\x12\n\x04type\x18\x01 \x01(\tR\x04type\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n\x04\x62ody\x18\x03 \x01(\tR\x04\x62ody\x12\x1d\n\ndoc_string\x18\x04 \x01(\tR\tdocString\x12\x42\n\x05\x61ttrs\x18\x05 \x03(\x0b\x32,.mplang.protos.v1alpha1.FuncProto.AttrsEntryR\x05\x61ttrs\x1a[\n\nAttrsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32!.mplang.protos.v1alpha1.AttrProtoR\x05value:\x02\x38\x01\"h\n\x0fTensorTypeProto\x12\x36\n\x05\x64type\x18\x01 \x01(\x0e\x32 .mplang.protos.v1alpha1.DataTypeR\x05\x64type\x12\x1d\n\nshape_dims\x18\x02 \x03(\x03R\tshapeDims\"\xaf\x01\n\x0eTableTypeProto\x12G\n\x07\x63olumns\x18\x01 \x03(\x0b\x32-.mplang.protos.v1alpha1.TableTypeProto.ColumnR\x07\x63olumns\x1aT\n\x06\x43olumn\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x36\n\x05\x64type\x18\x02 \x01(\x0e\x32 .mplang.protos.v1alpha1.DataTypeR\x05\x64type\"\xe8\x02\n\x0bMPTypeProto\x12J\n\x0btensor_type\x18\x01 \x01(\x0b\x32\'.mplang.protos.v1alpha1.TensorTypeProtoH\x00R\ntensorType\x12G\n\ntable_type\x18\x02 \x01(\x0b\x32&.mplang.protos.v1alpha1.TableTypeProtoH\x00R\ttableType\x12\x14\n\x05pmask\x18\x03 \x01(\x03R\x05pmask\x12\x44\n\x05\x61ttrs\x18\x04 \x03(\x0b\x32..mplang.protos.v1alpha1.MPTypeProto.AttrsEntryR\x05\x61ttrs\x1a[\n\nAttrsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32!.mplang.protos.v1alpha1.AttrProtoR\x05value:\x02\x38\x01\x42\x0b\n\ttype_spec\"\xd2\x02\n\tNodeProto\x12\x17\n\x07op_type\x18\x01 \x01(\tR\x06opType\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x16\n\x06inputs\x18\x03 \x03(\tR\x06inputs\x12@\n\touts_info\x18\x05 \x03(\x0b\x32#.mplang.protos.v1alpha1.MPTypeProtoR\x08outsInfo\x12\x42\n\x05\x61ttrs\x18\x04 \x03(\x0b\x32,.mplang.protos.v1alpha1.NodeProto.AttrsEntryR\x05\x61ttrs\x12\x1d\n\ndoc_string\x18\x06 \x01(\tR\tdocString\x1a[\n\nAttrsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32!.mplang.protos.v1alpha1.AttrProtoR\x05value:\x02\x38\x01\"e\n\x0bVersionInfo\x12\x14\n\x05major\x18\x01 \x01(\x05R\x05major\x12\x14\n\x05minor\x18\x02 \x01(\x05R\x05minor\x12\x14\n\x05patch\x18\x03 \x01(\x05R\x05patch\x12\x14\n\x05\x65xtra\x18\x04 \x01(\tR\x05\x65xtra\"\xc0\x02\n\nGraphProto\x12=\n\x07version\x18\x01 \x01(\x0b\x32#.mplang.protos.v1alpha1.VersionInfoR\x07version\x12\x37\n\x05nodes\x18\x02 \x03(\x0b\x32!.mplang.protos.v1alpha1.NodeProtoR\x05nodes\x12\x18\n\x07outputs\x18\x03 \x03(\tR\x07outputs\x12\x43\n\x05\x61ttrs\x18\x04 \x03(\x0b\x32-.mplang.protos.v1alpha1.GraphProto.AttrsEntryR\x05\x61ttrs\x1a[\n\nAttrsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32!.mplang.protos.v1alpha1.AttrProtoR\x05value:\x02\x38\x01*\xea\x01\n\x08\x44\x61taType\x12\r\n\tUNDEFINED\x10\x00\x12\x06\n\x02U8\x10\x01\x12\x06\n\x02I8\x10\x02\x12\x07\n\x03U16\x10\x03\x12\x07\n\x03I16\x10\x04\x12\x07\n\x03U32\x10\x05\x12\x07\n\x03I32\x10\x06\x12\x07\n\x03U64\x10\x07\x12\x07\n\x03I64\x10\x08\x12\x07\n\x03\x46\x31\x36\x10\t\x12\x07\n\x03\x46\x33\x32\x10\n\x12\x07\n\x03\x46\x36\x34\x10\x0b\x12\n\n\x06STRING\x10\x0c\x12\x08\n\x04\x42OOL\x10\r\x12\r\n\tCOMPLEX64\x10\x0e\x12\x0e\n\nCOMPLEX128\x10\x0f\x12\x0c\n\x08\x42\x46LOAT16\x10\x10\x12\x08\n\x04\x44\x41TE\x10\x11\x12\x08\n\x04TIME\x10\x12\x12\r\n\tTIMESTAMP\x10\x13\x12\x08\n\x04JSON\x10\x14\x42.Z,github.com/secretflow/mplang/protos/v1alpha1b\x06proto3')
|
|
18
|
-
|
|
19
|
-
_globals = globals()
|
|
20
|
-
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
21
|
-
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mplang.protos.v1alpha1.mpir_pb2', _globals)
|
|
22
|
-
if not _descriptor._USE_C_DESCRIPTORS:
|
|
23
|
-
_globals['DESCRIPTOR']._loaded_options = None
|
|
24
|
-
_globals['DESCRIPTOR']._serialized_options = b'Z,github.com/secretflow/mplang/protos/v1alpha1'
|
|
25
|
-
_globals['_FUNCPROTO_ATTRSENTRY']._loaded_options = None
|
|
26
|
-
_globals['_FUNCPROTO_ATTRSENTRY']._serialized_options = b'8\001'
|
|
27
|
-
_globals['_MPTYPEPROTO_ATTRSENTRY']._loaded_options = None
|
|
28
|
-
_globals['_MPTYPEPROTO_ATTRSENTRY']._serialized_options = b'8\001'
|
|
29
|
-
_globals['_NODEPROTO_ATTRSENTRY']._loaded_options = None
|
|
30
|
-
_globals['_NODEPROTO_ATTRSENTRY']._serialized_options = b'8\001'
|
|
31
|
-
_globals['_GRAPHPROTO_ATTRSENTRY']._loaded_options = None
|
|
32
|
-
_globals['_GRAPHPROTO_ATTRSENTRY']._serialized_options = b'8\001'
|
|
33
|
-
_globals['_DATATYPE']._serialized_start=2223
|
|
34
|
-
_globals['_DATATYPE']._serialized_end=2457
|
|
35
|
-
_globals['_ATTRPROTO']._serialized_start=62
|
|
36
|
-
_globals['_ATTRPROTO']._serialized_end=540
|
|
37
|
-
_globals['_ATTRPROTO_ATTRTYPE']._serialized_start=402
|
|
38
|
-
_globals['_ATTRPROTO_ATTRTYPE']._serialized_end=540
|
|
39
|
-
_globals['_FUNCPROTO']._serialized_start=543
|
|
40
|
-
_globals['_FUNCPROTO']._serialized_end=806
|
|
41
|
-
_globals['_FUNCPROTO_ATTRSENTRY']._serialized_start=715
|
|
42
|
-
_globals['_FUNCPROTO_ATTRSENTRY']._serialized_end=806
|
|
43
|
-
_globals['_TENSORTYPEPROTO']._serialized_start=808
|
|
44
|
-
_globals['_TENSORTYPEPROTO']._serialized_end=912
|
|
45
|
-
_globals['_TABLETYPEPROTO']._serialized_start=915
|
|
46
|
-
_globals['_TABLETYPEPROTO']._serialized_end=1090
|
|
47
|
-
_globals['_TABLETYPEPROTO_COLUMN']._serialized_start=1006
|
|
48
|
-
_globals['_TABLETYPEPROTO_COLUMN']._serialized_end=1090
|
|
49
|
-
_globals['_MPTYPEPROTO']._serialized_start=1093
|
|
50
|
-
_globals['_MPTYPEPROTO']._serialized_end=1453
|
|
51
|
-
_globals['_MPTYPEPROTO_ATTRSENTRY']._serialized_start=715
|
|
52
|
-
_globals['_MPTYPEPROTO_ATTRSENTRY']._serialized_end=806
|
|
53
|
-
_globals['_NODEPROTO']._serialized_start=1456
|
|
54
|
-
_globals['_NODEPROTO']._serialized_end=1794
|
|
55
|
-
_globals['_NODEPROTO_ATTRSENTRY']._serialized_start=715
|
|
56
|
-
_globals['_NODEPROTO_ATTRSENTRY']._serialized_end=806
|
|
57
|
-
_globals['_VERSIONINFO']._serialized_start=1796
|
|
58
|
-
_globals['_VERSIONINFO']._serialized_end=1897
|
|
59
|
-
_globals['_GRAPHPROTO']._serialized_start=1900
|
|
60
|
-
_globals['_GRAPHPROTO']._serialized_end=2220
|
|
61
|
-
_globals['_GRAPHPROTO_ATTRSENTRY']._serialized_start=715
|
|
62
|
-
_globals['_GRAPHPROTO_ATTRSENTRY']._serialized_end=806
|
|
63
|
-
# @@protoc_insertion_point(module_scope)
|
|
@@ -1,557 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
@generated by mypy-protobuf. Do not edit manually!
|
|
3
|
-
isort:skip_file
|
|
4
|
-
Copyright 2025 Ant Group Co., Ltd.
|
|
5
|
-
|
|
6
|
-
Licensed under the Apache License, Version 2.0 (the "License");
|
|
7
|
-
you may not use this file except in compliance with the License.
|
|
8
|
-
You may obtain a copy of the License at
|
|
9
|
-
|
|
10
|
-
http://www.apache.org/licenses/LICENSE-2.0
|
|
11
|
-
|
|
12
|
-
Unless required by applicable law or agreed to in writing, software
|
|
13
|
-
distributed under the License is distributed on an "AS IS" BASIS,
|
|
14
|
-
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
15
|
-
See the License for the specific language governing permissions and
|
|
16
|
-
limitations under the License.
|
|
17
|
-
"""
|
|
18
|
-
import builtins
|
|
19
|
-
import collections.abc
|
|
20
|
-
import google.protobuf.descriptor
|
|
21
|
-
import google.protobuf.internal.containers
|
|
22
|
-
import google.protobuf.internal.enum_type_wrapper
|
|
23
|
-
import google.protobuf.message
|
|
24
|
-
import sys
|
|
25
|
-
import typing
|
|
26
|
-
|
|
27
|
-
if sys.version_info >= (3, 10):
|
|
28
|
-
import typing as typing_extensions
|
|
29
|
-
else:
|
|
30
|
-
import typing_extensions
|
|
31
|
-
|
|
32
|
-
DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
|
|
33
|
-
|
|
34
|
-
class _DataType:
|
|
35
|
-
ValueType = typing.NewType("ValueType", builtins.int)
|
|
36
|
-
V: typing_extensions.TypeAlias = ValueType
|
|
37
|
-
|
|
38
|
-
class _DataTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DataType.ValueType], builtins.type):
|
|
39
|
-
DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
|
|
40
|
-
UNDEFINED: _DataType.ValueType # 0
|
|
41
|
-
"""Undefined data type"""
|
|
42
|
-
U8: _DataType.ValueType # 1
|
|
43
|
-
"""uint8_t"""
|
|
44
|
-
I8: _DataType.ValueType # 2
|
|
45
|
-
"""int8_t"""
|
|
46
|
-
U16: _DataType.ValueType # 3
|
|
47
|
-
"""uint16_t"""
|
|
48
|
-
I16: _DataType.ValueType # 4
|
|
49
|
-
"""int16_t"""
|
|
50
|
-
U32: _DataType.ValueType # 5
|
|
51
|
-
"""uint32_t"""
|
|
52
|
-
I32: _DataType.ValueType # 6
|
|
53
|
-
"""int32_t"""
|
|
54
|
-
U64: _DataType.ValueType # 7
|
|
55
|
-
"""uint64_t"""
|
|
56
|
-
I64: _DataType.ValueType # 8
|
|
57
|
-
"""int64_t"""
|
|
58
|
-
F16: _DataType.ValueType # 9
|
|
59
|
-
"""float16"""
|
|
60
|
-
F32: _DataType.ValueType # 10
|
|
61
|
-
"""float32"""
|
|
62
|
-
F64: _DataType.ValueType # 11
|
|
63
|
-
"""double"""
|
|
64
|
-
STRING: _DataType.ValueType # 12
|
|
65
|
-
"""string (table only)"""
|
|
66
|
-
BOOL: _DataType.ValueType # 13
|
|
67
|
-
"""bool"""
|
|
68
|
-
COMPLEX64: _DataType.ValueType # 14
|
|
69
|
-
"""complex with float32 real and imaginary components"""
|
|
70
|
-
COMPLEX128: _DataType.ValueType # 15
|
|
71
|
-
"""complex with float64 real and imaginary components"""
|
|
72
|
-
BFLOAT16: _DataType.ValueType # 16
|
|
73
|
-
"""bfloat16"""
|
|
74
|
-
DATE: _DataType.ValueType # 17
|
|
75
|
-
"""date (table only)"""
|
|
76
|
-
TIME: _DataType.ValueType # 18
|
|
77
|
-
"""time (table only)"""
|
|
78
|
-
TIMESTAMP: _DataType.ValueType # 19
|
|
79
|
-
"""timestamp (table only)"""
|
|
80
|
-
JSON: _DataType.ValueType # 20
|
|
81
|
-
"""json (table only)"""
|
|
82
|
-
|
|
83
|
-
class DataType(_DataType, metaclass=_DataTypeEnumTypeWrapper):
|
|
84
|
-
"""Data type enumeration"""
|
|
85
|
-
|
|
86
|
-
UNDEFINED: DataType.ValueType # 0
|
|
87
|
-
"""Undefined data type"""
|
|
88
|
-
U8: DataType.ValueType # 1
|
|
89
|
-
"""uint8_t"""
|
|
90
|
-
I8: DataType.ValueType # 2
|
|
91
|
-
"""int8_t"""
|
|
92
|
-
U16: DataType.ValueType # 3
|
|
93
|
-
"""uint16_t"""
|
|
94
|
-
I16: DataType.ValueType # 4
|
|
95
|
-
"""int16_t"""
|
|
96
|
-
U32: DataType.ValueType # 5
|
|
97
|
-
"""uint32_t"""
|
|
98
|
-
I32: DataType.ValueType # 6
|
|
99
|
-
"""int32_t"""
|
|
100
|
-
U64: DataType.ValueType # 7
|
|
101
|
-
"""uint64_t"""
|
|
102
|
-
I64: DataType.ValueType # 8
|
|
103
|
-
"""int64_t"""
|
|
104
|
-
F16: DataType.ValueType # 9
|
|
105
|
-
"""float16"""
|
|
106
|
-
F32: DataType.ValueType # 10
|
|
107
|
-
"""float32"""
|
|
108
|
-
F64: DataType.ValueType # 11
|
|
109
|
-
"""double"""
|
|
110
|
-
STRING: DataType.ValueType # 12
|
|
111
|
-
"""string (table only)"""
|
|
112
|
-
BOOL: DataType.ValueType # 13
|
|
113
|
-
"""bool"""
|
|
114
|
-
COMPLEX64: DataType.ValueType # 14
|
|
115
|
-
"""complex with float32 real and imaginary components"""
|
|
116
|
-
COMPLEX128: DataType.ValueType # 15
|
|
117
|
-
"""complex with float64 real and imaginary components"""
|
|
118
|
-
BFLOAT16: DataType.ValueType # 16
|
|
119
|
-
"""bfloat16"""
|
|
120
|
-
DATE: DataType.ValueType # 17
|
|
121
|
-
"""date (table only)"""
|
|
122
|
-
TIME: DataType.ValueType # 18
|
|
123
|
-
"""time (table only)"""
|
|
124
|
-
TIMESTAMP: DataType.ValueType # 19
|
|
125
|
-
"""timestamp (table only)"""
|
|
126
|
-
JSON: DataType.ValueType # 20
|
|
127
|
-
"""json (table only)"""
|
|
128
|
-
global___DataType = DataType
|
|
129
|
-
|
|
130
|
-
@typing_extensions.final
|
|
131
|
-
class AttrProto(google.protobuf.message.Message):
|
|
132
|
-
"""Custom attribute message (without name field since it's the map key)"""
|
|
133
|
-
|
|
134
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
135
|
-
|
|
136
|
-
class _AttrType:
|
|
137
|
-
ValueType = typing.NewType("ValueType", builtins.int)
|
|
138
|
-
V: typing_extensions.TypeAlias = ValueType
|
|
139
|
-
|
|
140
|
-
class _AttrTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AttrProto._AttrType.ValueType], builtins.type):
|
|
141
|
-
DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
|
|
142
|
-
UNDEFINED: AttrProto._AttrType.ValueType # 0
|
|
143
|
-
"""Undefined attribute type"""
|
|
144
|
-
FLOAT: AttrProto._AttrType.ValueType # 1
|
|
145
|
-
"""Single float value"""
|
|
146
|
-
INT: AttrProto._AttrType.ValueType # 2
|
|
147
|
-
"""Single integer value"""
|
|
148
|
-
STRING: AttrProto._AttrType.ValueType # 3
|
|
149
|
-
"""Single string value"""
|
|
150
|
-
BOOL: AttrProto._AttrType.ValueType # 4
|
|
151
|
-
"""Single boolean value"""
|
|
152
|
-
BYTES: AttrProto._AttrType.ValueType # 5
|
|
153
|
-
"""Binary data"""
|
|
154
|
-
FLOATS: AttrProto._AttrType.ValueType # 6
|
|
155
|
-
"""Array of float values"""
|
|
156
|
-
INTS: AttrProto._AttrType.ValueType # 7
|
|
157
|
-
"""Array of integer values"""
|
|
158
|
-
STRINGS: AttrProto._AttrType.ValueType # 8
|
|
159
|
-
"""Array of string values"""
|
|
160
|
-
FUNCTION: AttrProto._AttrType.ValueType # 10
|
|
161
|
-
"""Textual function reference"""
|
|
162
|
-
GRAPH: AttrProto._AttrType.ValueType # 11
|
|
163
|
-
"""Nested function graph"""
|
|
164
|
-
|
|
165
|
-
class AttrType(_AttrType, metaclass=_AttrTypeEnumTypeWrapper):
|
|
166
|
-
"""Define possible attribute types"""
|
|
167
|
-
|
|
168
|
-
UNDEFINED: AttrProto.AttrType.ValueType # 0
|
|
169
|
-
"""Undefined attribute type"""
|
|
170
|
-
FLOAT: AttrProto.AttrType.ValueType # 1
|
|
171
|
-
"""Single float value"""
|
|
172
|
-
INT: AttrProto.AttrType.ValueType # 2
|
|
173
|
-
"""Single integer value"""
|
|
174
|
-
STRING: AttrProto.AttrType.ValueType # 3
|
|
175
|
-
"""Single string value"""
|
|
176
|
-
BOOL: AttrProto.AttrType.ValueType # 4
|
|
177
|
-
"""Single boolean value"""
|
|
178
|
-
BYTES: AttrProto.AttrType.ValueType # 5
|
|
179
|
-
"""Binary data"""
|
|
180
|
-
FLOATS: AttrProto.AttrType.ValueType # 6
|
|
181
|
-
"""Array of float values"""
|
|
182
|
-
INTS: AttrProto.AttrType.ValueType # 7
|
|
183
|
-
"""Array of integer values"""
|
|
184
|
-
STRINGS: AttrProto.AttrType.ValueType # 8
|
|
185
|
-
"""Array of string values"""
|
|
186
|
-
FUNCTION: AttrProto.AttrType.ValueType # 10
|
|
187
|
-
"""Textual function reference"""
|
|
188
|
-
GRAPH: AttrProto.AttrType.ValueType # 11
|
|
189
|
-
"""Nested function graph"""
|
|
190
|
-
|
|
191
|
-
TYPE_FIELD_NUMBER: builtins.int
|
|
192
|
-
F_FIELD_NUMBER: builtins.int
|
|
193
|
-
I_FIELD_NUMBER: builtins.int
|
|
194
|
-
S_FIELD_NUMBER: builtins.int
|
|
195
|
-
B_FIELD_NUMBER: builtins.int
|
|
196
|
-
RAW_BYTES_FIELD_NUMBER: builtins.int
|
|
197
|
-
FLOATS_FIELD_NUMBER: builtins.int
|
|
198
|
-
INTS_FIELD_NUMBER: builtins.int
|
|
199
|
-
STRS_FIELD_NUMBER: builtins.int
|
|
200
|
-
FUNC_FIELD_NUMBER: builtins.int
|
|
201
|
-
GRAPH_FIELD_NUMBER: builtins.int
|
|
202
|
-
type: global___AttrProto.AttrType.ValueType
|
|
203
|
-
"""Type of the attribute"""
|
|
204
|
-
f: builtins.float
|
|
205
|
-
"""FLOAT value"""
|
|
206
|
-
i: builtins.int
|
|
207
|
-
"""INT value"""
|
|
208
|
-
s: builtins.str
|
|
209
|
-
"""STRING value"""
|
|
210
|
-
b: builtins.bool
|
|
211
|
-
"""BOOL value"""
|
|
212
|
-
raw_bytes: builtins.bytes
|
|
213
|
-
"""BYTES - for raw binary data"""
|
|
214
|
-
@property
|
|
215
|
-
def floats(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]:
|
|
216
|
-
"""FLOATS - array of float values"""
|
|
217
|
-
@property
|
|
218
|
-
def ints(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
|
|
219
|
-
"""INTS - array of integer values"""
|
|
220
|
-
@property
|
|
221
|
-
def strs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
|
|
222
|
-
"""STRINGS - array of string values"""
|
|
223
|
-
@property
|
|
224
|
-
def func(self) -> global___FuncProto:
|
|
225
|
-
"""FUNCTION - textual function reference"""
|
|
226
|
-
@property
|
|
227
|
-
def graph(self) -> global___GraphProto:
|
|
228
|
-
"""GRAPH - nested function graph"""
|
|
229
|
-
def __init__(
|
|
230
|
-
self,
|
|
231
|
-
*,
|
|
232
|
-
type: global___AttrProto.AttrType.ValueType = ...,
|
|
233
|
-
f: builtins.float = ...,
|
|
234
|
-
i: builtins.int = ...,
|
|
235
|
-
s: builtins.str = ...,
|
|
236
|
-
b: builtins.bool = ...,
|
|
237
|
-
raw_bytes: builtins.bytes = ...,
|
|
238
|
-
floats: collections.abc.Iterable[builtins.float] | None = ...,
|
|
239
|
-
ints: collections.abc.Iterable[builtins.int] | None = ...,
|
|
240
|
-
strs: collections.abc.Iterable[builtins.str] | None = ...,
|
|
241
|
-
func: global___FuncProto | None = ...,
|
|
242
|
-
graph: global___GraphProto | None = ...,
|
|
243
|
-
) -> None: ...
|
|
244
|
-
def HasField(self, field_name: typing_extensions.Literal["func", b"func", "graph", b"graph"]) -> builtins.bool: ...
|
|
245
|
-
def ClearField(self, field_name: typing_extensions.Literal["b", b"b", "f", b"f", "floats", b"floats", "func", b"func", "graph", b"graph", "i", b"i", "ints", b"ints", "raw_bytes", b"raw_bytes", "s", b"s", "strs", b"strs", "type", b"type"]) -> None: ...
|
|
246
|
-
|
|
247
|
-
global___AttrProto = AttrProto
|
|
248
|
-
|
|
249
|
-
@typing_extensions.final
|
|
250
|
-
class FuncProto(google.protobuf.message.Message):
|
|
251
|
-
"""Function prototype message"""
|
|
252
|
-
|
|
253
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
254
|
-
|
|
255
|
-
@typing_extensions.final
|
|
256
|
-
class AttrsEntry(google.protobuf.message.Message):
|
|
257
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
258
|
-
|
|
259
|
-
KEY_FIELD_NUMBER: builtins.int
|
|
260
|
-
VALUE_FIELD_NUMBER: builtins.int
|
|
261
|
-
key: builtins.str
|
|
262
|
-
@property
|
|
263
|
-
def value(self) -> global___AttrProto: ...
|
|
264
|
-
def __init__(
|
|
265
|
-
self,
|
|
266
|
-
*,
|
|
267
|
-
key: builtins.str = ...,
|
|
268
|
-
value: global___AttrProto | None = ...,
|
|
269
|
-
) -> None: ...
|
|
270
|
-
def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
|
|
271
|
-
def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
|
|
272
|
-
|
|
273
|
-
TYPE_FIELD_NUMBER: builtins.int
|
|
274
|
-
NAME_FIELD_NUMBER: builtins.int
|
|
275
|
-
BODY_FIELD_NUMBER: builtins.int
|
|
276
|
-
DOC_STRING_FIELD_NUMBER: builtins.int
|
|
277
|
-
ATTRS_FIELD_NUMBER: builtins.int
|
|
278
|
-
type: builtins.str
|
|
279
|
-
"""Function type"""
|
|
280
|
-
name: builtins.str
|
|
281
|
-
"""Function name"""
|
|
282
|
-
body: builtins.str
|
|
283
|
-
"""Function body as a string (e.g., Python code)"""
|
|
284
|
-
doc_string: builtins.str
|
|
285
|
-
"""Documentation string"""
|
|
286
|
-
@property
|
|
287
|
-
def attrs(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___AttrProto]:
|
|
288
|
-
"""Function attributes"""
|
|
289
|
-
def __init__(
|
|
290
|
-
self,
|
|
291
|
-
*,
|
|
292
|
-
type: builtins.str = ...,
|
|
293
|
-
name: builtins.str = ...,
|
|
294
|
-
body: builtins.str = ...,
|
|
295
|
-
doc_string: builtins.str = ...,
|
|
296
|
-
attrs: collections.abc.Mapping[builtins.str, global___AttrProto] | None = ...,
|
|
297
|
-
) -> None: ...
|
|
298
|
-
def ClearField(self, field_name: typing_extensions.Literal["attrs", b"attrs", "body", b"body", "doc_string", b"doc_string", "name", b"name", "type", b"type"]) -> None: ...
|
|
299
|
-
|
|
300
|
-
global___FuncProto = FuncProto
|
|
301
|
-
|
|
302
|
-
@typing_extensions.final
|
|
303
|
-
class TensorTypeProto(google.protobuf.message.Message):
|
|
304
|
-
"""Tensor type definition"""
|
|
305
|
-
|
|
306
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
307
|
-
|
|
308
|
-
DTYPE_FIELD_NUMBER: builtins.int
|
|
309
|
-
SHAPE_DIMS_FIELD_NUMBER: builtins.int
|
|
310
|
-
dtype: global___DataType.ValueType
|
|
311
|
-
"""Data type of the tensor elements"""
|
|
312
|
-
@property
|
|
313
|
-
def shape_dims(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
|
|
314
|
-
"""Shape dimensions of the tensor"""
|
|
315
|
-
def __init__(
|
|
316
|
-
self,
|
|
317
|
-
*,
|
|
318
|
-
dtype: global___DataType.ValueType = ...,
|
|
319
|
-
shape_dims: collections.abc.Iterable[builtins.int] | None = ...,
|
|
320
|
-
) -> None: ...
|
|
321
|
-
def ClearField(self, field_name: typing_extensions.Literal["dtype", b"dtype", "shape_dims", b"shape_dims"]) -> None: ...
|
|
322
|
-
|
|
323
|
-
global___TensorTypeProto = TensorTypeProto
|
|
324
|
-
|
|
325
|
-
@typing_extensions.final
|
|
326
|
-
class TableTypeProto(google.protobuf.message.Message):
|
|
327
|
-
"""Table type definition"""
|
|
328
|
-
|
|
329
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
330
|
-
|
|
331
|
-
@typing_extensions.final
|
|
332
|
-
class Column(google.protobuf.message.Message):
|
|
333
|
-
"""Column definition within a table"""
|
|
334
|
-
|
|
335
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
336
|
-
|
|
337
|
-
NAME_FIELD_NUMBER: builtins.int
|
|
338
|
-
DTYPE_FIELD_NUMBER: builtins.int
|
|
339
|
-
name: builtins.str
|
|
340
|
-
"""Name of the column"""
|
|
341
|
-
dtype: global___DataType.ValueType
|
|
342
|
-
"""Data type of the column"""
|
|
343
|
-
def __init__(
|
|
344
|
-
self,
|
|
345
|
-
*,
|
|
346
|
-
name: builtins.str = ...,
|
|
347
|
-
dtype: global___DataType.ValueType = ...,
|
|
348
|
-
) -> None: ...
|
|
349
|
-
def ClearField(self, field_name: typing_extensions.Literal["dtype", b"dtype", "name", b"name"]) -> None: ...
|
|
350
|
-
|
|
351
|
-
COLUMNS_FIELD_NUMBER: builtins.int
|
|
352
|
-
@property
|
|
353
|
-
def columns(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TableTypeProto.Column]:
|
|
354
|
-
"""List of columns in the table"""
|
|
355
|
-
def __init__(
|
|
356
|
-
self,
|
|
357
|
-
*,
|
|
358
|
-
columns: collections.abc.Iterable[global___TableTypeProto.Column] | None = ...,
|
|
359
|
-
) -> None: ...
|
|
360
|
-
def ClearField(self, field_name: typing_extensions.Literal["columns", b"columns"]) -> None: ...
|
|
361
|
-
|
|
362
|
-
global___TableTypeProto = TableTypeProto
|
|
363
|
-
|
|
364
|
-
@typing_extensions.final
|
|
365
|
-
class MPTypeProto(google.protobuf.message.Message):
|
|
366
|
-
"""Multi-party type definition"""
|
|
367
|
-
|
|
368
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
369
|
-
|
|
370
|
-
@typing_extensions.final
|
|
371
|
-
class AttrsEntry(google.protobuf.message.Message):
|
|
372
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
373
|
-
|
|
374
|
-
KEY_FIELD_NUMBER: builtins.int
|
|
375
|
-
VALUE_FIELD_NUMBER: builtins.int
|
|
376
|
-
key: builtins.str
|
|
377
|
-
@property
|
|
378
|
-
def value(self) -> global___AttrProto: ...
|
|
379
|
-
def __init__(
|
|
380
|
-
self,
|
|
381
|
-
*,
|
|
382
|
-
key: builtins.str = ...,
|
|
383
|
-
value: global___AttrProto | None = ...,
|
|
384
|
-
) -> None: ...
|
|
385
|
-
def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
|
|
386
|
-
def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
|
|
387
|
-
|
|
388
|
-
TENSOR_TYPE_FIELD_NUMBER: builtins.int
|
|
389
|
-
TABLE_TYPE_FIELD_NUMBER: builtins.int
|
|
390
|
-
PMASK_FIELD_NUMBER: builtins.int
|
|
391
|
-
ATTRS_FIELD_NUMBER: builtins.int
|
|
392
|
-
@property
|
|
393
|
-
def tensor_type(self) -> global___TensorTypeProto:
|
|
394
|
-
"""Tensor type specification"""
|
|
395
|
-
@property
|
|
396
|
-
def table_type(self) -> global___TableTypeProto:
|
|
397
|
-
"""Table type specification"""
|
|
398
|
-
pmask: builtins.int
|
|
399
|
-
"""Party mask (-1 for dynamic mask, >=0 for static mask)"""
|
|
400
|
-
@property
|
|
401
|
-
def attrs(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___AttrProto]:
|
|
402
|
-
"""Additional attributes"""
|
|
403
|
-
def __init__(
|
|
404
|
-
self,
|
|
405
|
-
*,
|
|
406
|
-
tensor_type: global___TensorTypeProto | None = ...,
|
|
407
|
-
table_type: global___TableTypeProto | None = ...,
|
|
408
|
-
pmask: builtins.int = ...,
|
|
409
|
-
attrs: collections.abc.Mapping[builtins.str, global___AttrProto] | None = ...,
|
|
410
|
-
) -> None: ...
|
|
411
|
-
def HasField(self, field_name: typing_extensions.Literal["table_type", b"table_type", "tensor_type", b"tensor_type", "type_spec", b"type_spec"]) -> builtins.bool: ...
|
|
412
|
-
def ClearField(self, field_name: typing_extensions.Literal["attrs", b"attrs", "pmask", b"pmask", "table_type", b"table_type", "tensor_type", b"tensor_type", "type_spec", b"type_spec"]) -> None: ...
|
|
413
|
-
def WhichOneof(self, oneof_group: typing_extensions.Literal["type_spec", b"type_spec"]) -> typing_extensions.Literal["tensor_type", "table_type"] | None: ...
|
|
414
|
-
|
|
415
|
-
global___MPTypeProto = MPTypeProto
|
|
416
|
-
|
|
417
|
-
@typing_extensions.final
|
|
418
|
-
class NodeProto(google.protobuf.message.Message):
|
|
419
|
-
"""Node prototype definition"""
|
|
420
|
-
|
|
421
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
422
|
-
|
|
423
|
-
@typing_extensions.final
|
|
424
|
-
class AttrsEntry(google.protobuf.message.Message):
|
|
425
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
426
|
-
|
|
427
|
-
KEY_FIELD_NUMBER: builtins.int
|
|
428
|
-
VALUE_FIELD_NUMBER: builtins.int
|
|
429
|
-
key: builtins.str
|
|
430
|
-
@property
|
|
431
|
-
def value(self) -> global___AttrProto: ...
|
|
432
|
-
def __init__(
|
|
433
|
-
self,
|
|
434
|
-
*,
|
|
435
|
-
key: builtins.str = ...,
|
|
436
|
-
value: global___AttrProto | None = ...,
|
|
437
|
-
) -> None: ...
|
|
438
|
-
def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
|
|
439
|
-
def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
|
|
440
|
-
|
|
441
|
-
OP_TYPE_FIELD_NUMBER: builtins.int
|
|
442
|
-
NAME_FIELD_NUMBER: builtins.int
|
|
443
|
-
INPUTS_FIELD_NUMBER: builtins.int
|
|
444
|
-
OUTS_INFO_FIELD_NUMBER: builtins.int
|
|
445
|
-
ATTRS_FIELD_NUMBER: builtins.int
|
|
446
|
-
DOC_STRING_FIELD_NUMBER: builtins.int
|
|
447
|
-
op_type: builtins.str
|
|
448
|
-
"""Operation type of the node"""
|
|
449
|
-
name: builtins.str
|
|
450
|
-
"""Name of the node"""
|
|
451
|
-
@property
|
|
452
|
-
def inputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
|
|
453
|
-
"""Input specifications in format {name:index}"""
|
|
454
|
-
@property
|
|
455
|
-
def outs_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MPTypeProto]:
|
|
456
|
-
"""Output type information"""
|
|
457
|
-
@property
|
|
458
|
-
def attrs(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___AttrProto]:
|
|
459
|
-
"""Node attributes"""
|
|
460
|
-
doc_string: builtins.str
|
|
461
|
-
"""Documentation string"""
|
|
462
|
-
def __init__(
|
|
463
|
-
self,
|
|
464
|
-
*,
|
|
465
|
-
op_type: builtins.str = ...,
|
|
466
|
-
name: builtins.str = ...,
|
|
467
|
-
inputs: collections.abc.Iterable[builtins.str] | None = ...,
|
|
468
|
-
outs_info: collections.abc.Iterable[global___MPTypeProto] | None = ...,
|
|
469
|
-
attrs: collections.abc.Mapping[builtins.str, global___AttrProto] | None = ...,
|
|
470
|
-
doc_string: builtins.str = ...,
|
|
471
|
-
) -> None: ...
|
|
472
|
-
def ClearField(self, field_name: typing_extensions.Literal["attrs", b"attrs", "doc_string", b"doc_string", "inputs", b"inputs", "name", b"name", "op_type", b"op_type", "outs_info", b"outs_info"]) -> None: ...
|
|
473
|
-
|
|
474
|
-
global___NodeProto = NodeProto
|
|
475
|
-
|
|
476
|
-
@typing_extensions.final
|
|
477
|
-
class VersionInfo(google.protobuf.message.Message):
|
|
478
|
-
"""Version information definition"""
|
|
479
|
-
|
|
480
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
481
|
-
|
|
482
|
-
MAJOR_FIELD_NUMBER: builtins.int
|
|
483
|
-
MINOR_FIELD_NUMBER: builtins.int
|
|
484
|
-
PATCH_FIELD_NUMBER: builtins.int
|
|
485
|
-
EXTRA_FIELD_NUMBER: builtins.int
|
|
486
|
-
major: builtins.int
|
|
487
|
-
"""Major version, incompatible changes"""
|
|
488
|
-
minor: builtins.int
|
|
489
|
-
"""Minor version, backward compatible changes"""
|
|
490
|
-
patch: builtins.int
|
|
491
|
-
"""Patch version, bug fixes"""
|
|
492
|
-
extra: builtins.str
|
|
493
|
-
"""Extra information, like "beta", "rc1", etc."""
|
|
494
|
-
def __init__(
|
|
495
|
-
self,
|
|
496
|
-
*,
|
|
497
|
-
major: builtins.int = ...,
|
|
498
|
-
minor: builtins.int = ...,
|
|
499
|
-
patch: builtins.int = ...,
|
|
500
|
-
extra: builtins.str = ...,
|
|
501
|
-
) -> None: ...
|
|
502
|
-
def ClearField(self, field_name: typing_extensions.Literal["extra", b"extra", "major", b"major", "minor", b"minor", "patch", b"patch"]) -> None: ...
|
|
503
|
-
|
|
504
|
-
global___VersionInfo = VersionInfo
|
|
505
|
-
|
|
506
|
-
@typing_extensions.final
|
|
507
|
-
class GraphProto(google.protobuf.message.Message):
|
|
508
|
-
"""Graph prototype definition"""
|
|
509
|
-
|
|
510
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
511
|
-
|
|
512
|
-
@typing_extensions.final
|
|
513
|
-
class AttrsEntry(google.protobuf.message.Message):
|
|
514
|
-
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
515
|
-
|
|
516
|
-
KEY_FIELD_NUMBER: builtins.int
|
|
517
|
-
VALUE_FIELD_NUMBER: builtins.int
|
|
518
|
-
key: builtins.str
|
|
519
|
-
@property
|
|
520
|
-
def value(self) -> global___AttrProto: ...
|
|
521
|
-
def __init__(
|
|
522
|
-
self,
|
|
523
|
-
*,
|
|
524
|
-
key: builtins.str = ...,
|
|
525
|
-
value: global___AttrProto | None = ...,
|
|
526
|
-
) -> None: ...
|
|
527
|
-
def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
|
|
528
|
-
def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
|
|
529
|
-
|
|
530
|
-
VERSION_FIELD_NUMBER: builtins.int
|
|
531
|
-
NODES_FIELD_NUMBER: builtins.int
|
|
532
|
-
OUTPUTS_FIELD_NUMBER: builtins.int
|
|
533
|
-
ATTRS_FIELD_NUMBER: builtins.int
|
|
534
|
-
@property
|
|
535
|
-
def version(self) -> global___VersionInfo:
|
|
536
|
-
"""Version information"""
|
|
537
|
-
@property
|
|
538
|
-
def nodes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NodeProto]:
|
|
539
|
-
"""List of nodes in the graph"""
|
|
540
|
-
@property
|
|
541
|
-
def outputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
|
|
542
|
-
"""Output specifications in format {name:index}"""
|
|
543
|
-
@property
|
|
544
|
-
def attrs(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___AttrProto]:
|
|
545
|
-
"""Graph attributes"""
|
|
546
|
-
def __init__(
|
|
547
|
-
self,
|
|
548
|
-
*,
|
|
549
|
-
version: global___VersionInfo | None = ...,
|
|
550
|
-
nodes: collections.abc.Iterable[global___NodeProto] | None = ...,
|
|
551
|
-
outputs: collections.abc.Iterable[builtins.str] | None = ...,
|
|
552
|
-
attrs: collections.abc.Mapping[builtins.str, global___AttrProto] | None = ...,
|
|
553
|
-
) -> None: ...
|
|
554
|
-
def HasField(self, field_name: typing_extensions.Literal["version", b"version"]) -> builtins.bool: ...
|
|
555
|
-
def ClearField(self, field_name: typing_extensions.Literal["attrs", b"attrs", "nodes", b"nodes", "outputs", b"outputs", "version", b"version"]) -> None: ...
|
|
556
|
-
|
|
557
|
-
global___GraphProto = GraphProto
|
|
@@ -1,34 +0,0 @@
|
|
|
1
|
-
# -*- coding: utf-8 -*-
|
|
2
|
-
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
-
# source: mplang/protos/v1alpha1/value.proto
|
|
4
|
-
# Protobuf Python Version: 5.26.1
|
|
5
|
-
"""Generated protocol buffer code."""
|
|
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
|
-
from google.protobuf.internal import builder as _builder
|
|
10
|
-
# @@protoc_insertion_point(imports)
|
|
11
|
-
|
|
12
|
-
_sym_db = _symbol_database.Default()
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"mplang/protos/v1alpha1/value.proto\x12\x16mplang.protos.v1alpha1\"\xe8\x02\n\x0eValueAttrProto\x12\x43\n\x04type\x18\x01 \x01(\x0e\x32/.mplang.protos.v1alpha1.ValueAttrProto.AttrTypeR\x04type\x12\x0c\n\x01\x66\x18\x02 \x01(\x02R\x01\x66\x12\x0c\n\x01i\x18\x03 \x01(\x03R\x01i\x12\x0c\n\x01s\x18\x04 \x01(\tR\x01s\x12\x0c\n\x01\x62\x18\x05 \x01(\x08R\x01\x62\x12\x1b\n\traw_bytes\x18\x06 \x01(\x0cR\x08rawBytes\x12\x16\n\x06\x66loats\x18\x07 \x03(\x02R\x06\x66loats\x12\x12\n\x04ints\x18\x08 \x03(\x03R\x04ints\x12\x12\n\x04strs\x18\t \x03(\tR\x04strs\"|\n\x08\x41ttrType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\n\n\x06STRING\x10\x03\x12\x08\n\x04\x42OOL\x10\x04\x12\t\n\x05\x42YTES\x10\x05\x12\n\n\x06\x46LOATS\x10\x06\x12\x08\n\x04INTS\x10\x07\x12\x0b\n\x07STRINGS\x10\x08\x12\t\n\x05\x45MPTY\x10\t\"\xa3\x02\n\nValueProto\x12\x12\n\x04kind\x18\x01 \x01(\tR\x04kind\x12#\n\rvalue_version\x18\x02 \x01(\rR\x0cvalueVersion\x12\x18\n\x07payload\x18\x03 \x01(\x0cR\x07payload\x12Y\n\rruntime_attrs\x18\x04 \x03(\x0b\x32\x34.mplang.protos.v1alpha1.ValueProto.RuntimeAttrsEntryR\x0cruntimeAttrs\x1ag\n\x11RuntimeAttrsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12<\n\x05value\x18\x02 \x01(\x0b\x32&.mplang.protos.v1alpha1.ValueAttrProtoR\x05value:\x02\x38\x01\x62\x06proto3')
|
|
18
|
-
|
|
19
|
-
_globals = globals()
|
|
20
|
-
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
21
|
-
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mplang.protos.v1alpha1.value_pb2', _globals)
|
|
22
|
-
if not _descriptor._USE_C_DESCRIPTORS:
|
|
23
|
-
DESCRIPTOR._loaded_options = None
|
|
24
|
-
_globals['_VALUEPROTO_RUNTIMEATTRSENTRY']._loaded_options = None
|
|
25
|
-
_globals['_VALUEPROTO_RUNTIMEATTRSENTRY']._serialized_options = b'8\001'
|
|
26
|
-
_globals['_VALUEATTRPROTO']._serialized_start=63
|
|
27
|
-
_globals['_VALUEATTRPROTO']._serialized_end=423
|
|
28
|
-
_globals['_VALUEATTRPROTO_ATTRTYPE']._serialized_start=299
|
|
29
|
-
_globals['_VALUEATTRPROTO_ATTRTYPE']._serialized_end=423
|
|
30
|
-
_globals['_VALUEPROTO']._serialized_start=426
|
|
31
|
-
_globals['_VALUEPROTO']._serialized_end=717
|
|
32
|
-
_globals['_VALUEPROTO_RUNTIMEATTRSENTRY']._serialized_start=614
|
|
33
|
-
_globals['_VALUEPROTO_RUNTIMEATTRSENTRY']._serialized_end=717
|
|
34
|
-
# @@protoc_insertion_point(module_scope)
|