{ "a_t128": { "inputs": { "input_ids": [ [ 1, 128 ], "int32" ], "attention_mask": [ [ 1, 128 ], "int32" ], "word_idx": [ [ 1, 128 ], "int32" ], "word_mask": [ [ 1, 128 ], "int32" ], "query_idx": [ [ 1, 32 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "cls_idx": [ [ 1, 32 ], "int32" ] }, "outputs": { "text_states": [ [ 1, 128, 768 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "cls_logits": [ [ 1, 32 ], "float16" ], "boundary_states": [ [ 1, 129, 128 ], "float16" ], "start_logits": [ [ 1, 32, 129 ], "float16" ], "end_logits": [ [ 1, 32, 129 ], "float16" ], "inside_logits": [ [ 1, 32, 128 ], "float16" ], "pool_start": [ [ 1, 129, 128 ], "float16" ], "pool_end": [ [ 1, 129, 128 ], "float16" ], "null_logits": [ [ 1, 32 ], "float16" ], "count_logits": [ [ 1, 32 ], "float16" ], "record_field": [ [ 1, 32, 128 ], "float16" ] } }, "b_t128": { "inputs": { "boundary_states": [ [ 1, 129, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 128, 768 ], "float16" ], "word_mask": [ [ 1, 128 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 129 ], "float16" ], "end_logits": [ [ 1, 32, 129 ], "float16" ], "inside_logits": [ [ 1, 32, 128 ], "float16" ], "pool_idx": [ [ 1, 192, 2 ], "int32" ], "pool_mask": [ [ 1, 192 ], "int32" ], "pool_prior": [ [ 1, 192 ], "float16" ] }, "outputs": { "pair_logits": [ [ 1, 192, 32 ], "float16" ], "record_cand": [ [ 1, 192, 128 ], "float16" ], "record_inst": [ [ 1, 192, 128 ], "float16" ], "latent_seed": [ [ 1, 192 ], "float16" ], "anchorless_inst": [ [ 1, 32, 128 ], "float16" ], "anchorless_object": [ [ 1, 32 ], "float16" ] } }, "c_t128": { "inputs": { "text_states": [ [ 1, 128, 768 ], "float16" ], "word_mask": [ [ 1, 128 ], "int32" ], "relation_query": [ [ 1, 8, 1536 ], "float16" ], "pairs": [ [ 1, 512, 4 ], "int32" ], "pair_relation": [ [ 1, 512 ], "int32" ], "pair_mask": [ [ 1, 512 ], "int32" ] }, "outputs": { "relation_logits": [ [ 1, 512 ], "float16" ] } }, "d_t128": { "inputs": { "boundary_states": [ [ 1, 129, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 128, 768 ], "float16" ], "word_mask": [ [ 1, 128 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 129 ], "float16" ], "end_logits": [ [ 1, 32, 129 ], "float16" ], "inside_logits": [ [ 1, 32, 128 ], "float16" ], "spans": [ [ 1, 64, 2 ], "int32" ], "span_mask": [ [ 1, 64 ], "int32" ] }, "outputs": { "explicit_logits": [ [ 1, 32, 64 ], "float16" ] } }, "a_t256": { "inputs": { "input_ids": [ [ 1, 256 ], "int32" ], "attention_mask": [ [ 1, 256 ], "int32" ], "word_idx": [ [ 1, 256 ], "int32" ], "word_mask": [ [ 1, 256 ], "int32" ], "query_idx": [ [ 1, 32 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "cls_idx": [ [ 1, 32 ], "int32" ] }, "outputs": { "text_states": [ [ 1, 256, 768 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "cls_logits": [ [ 1, 32 ], "float16" ], "boundary_states": [ [ 1, 257, 128 ], "float16" ], "start_logits": [ [ 1, 32, 257 ], "float16" ], "end_logits": [ [ 1, 32, 257 ], "float16" ], "inside_logits": [ [ 1, 32, 256 ], "float16" ], "pool_start": [ [ 1, 257, 128 ], "float16" ], "pool_end": [ [ 1, 257, 128 ], "float16" ], "null_logits": [ [ 1, 32 ], "float16" ], "count_logits": [ [ 1, 32 ], "float16" ], "record_field": [ [ 1, 32, 128 ], "float16" ] } }, "b_t256": { "inputs": { "boundary_states": [ [ 1, 257, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 256, 768 ], "float16" ], "word_mask": [ [ 1, 256 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 257 ], "float16" ], "end_logits": [ [ 1, 32, 257 ], "float16" ], "inside_logits": [ [ 1, 32, 256 ], "float16" ], "pool_idx": [ [ 1, 192, 2 ], "int32" ], "pool_mask": [ [ 1, 192 ], "int32" ], "pool_prior": [ [ 1, 192 ], "float16" ] }, "outputs": { "pair_logits": [ [ 1, 192, 32 ], "float16" ], "record_cand": [ [ 1, 192, 128 ], "float16" ], "record_inst": [ [ 1, 192, 128 ], "float16" ], "latent_seed": [ [ 1, 192 ], "float16" ], "anchorless_inst": [ [ 1, 32, 128 ], "float16" ], "anchorless_object": [ [ 1, 32 ], "float16" ] } }, "c_t256": { "inputs": { "text_states": [ [ 1, 256, 768 ], "float16" ], "word_mask": [ [ 1, 256 ], "int32" ], "relation_query": [ [ 1, 8, 1536 ], "float16" ], "pairs": [ [ 1, 512, 4 ], "int32" ], "pair_relation": [ [ 1, 512 ], "int32" ], "pair_mask": [ [ 1, 512 ], "int32" ] }, "outputs": { "relation_logits": [ [ 1, 512 ], "float16" ] } }, "d_t256": { "inputs": { "boundary_states": [ [ 1, 257, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 256, 768 ], "float16" ], "word_mask": [ [ 1, 256 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 257 ], "float16" ], "end_logits": [ [ 1, 32, 257 ], "float16" ], "inside_logits": [ [ 1, 32, 256 ], "float16" ], "spans": [ [ 1, 64, 2 ], "int32" ], "span_mask": [ [ 1, 64 ], "int32" ] }, "outputs": { "explicit_logits": [ [ 1, 32, 64 ], "float16" ] } }, "a_t512": { "inputs": { "input_ids": [ [ 1, 512 ], "int32" ], "attention_mask": [ [ 1, 512 ], "int32" ], "word_idx": [ [ 1, 512 ], "int32" ], "word_mask": [ [ 1, 512 ], "int32" ], "query_idx": [ [ 1, 32 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "cls_idx": [ [ 1, 32 ], "int32" ] }, "outputs": { "text_states": [ [ 1, 512, 768 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "cls_logits": [ [ 1, 32 ], "float16" ], "boundary_states": [ [ 1, 513, 128 ], "float16" ], "start_logits": [ [ 1, 32, 513 ], "float16" ], "end_logits": [ [ 1, 32, 513 ], "float16" ], "inside_logits": [ [ 1, 32, 512 ], "float16" ], "pool_start": [ [ 1, 513, 128 ], "float16" ], "pool_end": [ [ 1, 513, 128 ], "float16" ], "null_logits": [ [ 1, 32 ], "float16" ], "count_logits": [ [ 1, 32 ], "float16" ], "record_field": [ [ 1, 32, 128 ], "float16" ] } }, "b_t512": { "inputs": { "boundary_states": [ [ 1, 513, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 512, 768 ], "float16" ], "word_mask": [ [ 1, 512 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 513 ], "float16" ], "end_logits": [ [ 1, 32, 513 ], "float16" ], "inside_logits": [ [ 1, 32, 512 ], "float16" ], "pool_idx": [ [ 1, 192, 2 ], "int32" ], "pool_mask": [ [ 1, 192 ], "int32" ], "pool_prior": [ [ 1, 192 ], "float16" ] }, "outputs": { "pair_logits": [ [ 1, 192, 32 ], "float16" ], "record_cand": [ [ 1, 192, 128 ], "float16" ], "record_inst": [ [ 1, 192, 128 ], "float16" ], "latent_seed": [ [ 1, 192 ], "float16" ], "anchorless_inst": [ [ 1, 32, 128 ], "float16" ], "anchorless_object": [ [ 1, 32 ], "float16" ] } }, "c_t512": { "inputs": { "text_states": [ [ 1, 512, 768 ], "float16" ], "word_mask": [ [ 1, 512 ], "int32" ], "relation_query": [ [ 1, 8, 1536 ], "float16" ], "pairs": [ [ 1, 512, 4 ], "int32" ], "pair_relation": [ [ 1, 512 ], "int32" ], "pair_mask": [ [ 1, 512 ], "int32" ] }, "outputs": { "relation_logits": [ [ 1, 512 ], "float16" ] } }, "d_t512": { "inputs": { "boundary_states": [ [ 1, 513, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 512, 768 ], "float16" ], "word_mask": [ [ 1, 512 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 513 ], "float16" ], "end_logits": [ [ 1, 32, 513 ], "float16" ], "inside_logits": [ [ 1, 32, 512 ], "float16" ], "spans": [ [ 1, 64, 2 ], "int32" ], "span_mask": [ [ 1, 64 ], "int32" ] }, "outputs": { "explicit_logits": [ [ 1, 32, 64 ], "float16" ] } }, "a_t1024": { "inputs": { "input_ids": [ [ 1, 1024 ], "int32" ], "attention_mask": [ [ 1, 1024 ], "int32" ], "word_idx": [ [ 1, 1024 ], "int32" ], "word_mask": [ [ 1, 1024 ], "int32" ], "query_idx": [ [ 1, 32 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "cls_idx": [ [ 1, 32 ], "int32" ] }, "outputs": { "text_states": [ [ 1, 1024, 768 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "cls_logits": [ [ 1, 32 ], "float16" ], "boundary_states": [ [ 1, 1025, 128 ], "float16" ], "start_logits": [ [ 1, 32, 1025 ], "float16" ], "end_logits": [ [ 1, 32, 1025 ], "float16" ], "inside_logits": [ [ 1, 32, 1024 ], "float16" ], "pool_start": [ [ 1, 1025, 128 ], "float16" ], "pool_end": [ [ 1, 1025, 128 ], "float16" ], "null_logits": [ [ 1, 32 ], "float16" ], "count_logits": [ [ 1, 32 ], "float16" ], "record_field": [ [ 1, 32, 128 ], "float16" ] } }, "b_t1024": { "inputs": { "boundary_states": [ [ 1, 1025, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 1024, 768 ], "float16" ], "word_mask": [ [ 1, 1024 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 1025 ], "float16" ], "end_logits": [ [ 1, 32, 1025 ], "float16" ], "inside_logits": [ [ 1, 32, 1024 ], "float16" ], "pool_idx": [ [ 1, 192, 2 ], "int32" ], "pool_mask": [ [ 1, 192 ], "int32" ], "pool_prior": [ [ 1, 192 ], "float16" ] }, "outputs": { "pair_logits": [ [ 1, 192, 32 ], "float16" ], "record_cand": [ [ 1, 192, 128 ], "float16" ], "record_inst": [ [ 1, 192, 128 ], "float16" ], "latent_seed": [ [ 1, 192 ], "float16" ], "anchorless_inst": [ [ 1, 32, 128 ], "float16" ], "anchorless_object": [ [ 1, 32 ], "float16" ] } }, "c_t1024": { "inputs": { "text_states": [ [ 1, 1024, 768 ], "float16" ], "word_mask": [ [ 1, 1024 ], "int32" ], "relation_query": [ [ 1, 8, 1536 ], "float16" ], "pairs": [ [ 1, 512, 4 ], "int32" ], "pair_relation": [ [ 1, 512 ], "int32" ], "pair_mask": [ [ 1, 512 ], "int32" ] }, "outputs": { "relation_logits": [ [ 1, 512 ], "float16" ] } }, "d_t1024": { "inputs": { "boundary_states": [ [ 1, 1025, 128 ], "float16" ], "query_states": [ [ 1, 32, 768 ], "float16" ], "text_states": [ [ 1, 1024, 768 ], "float16" ], "word_mask": [ [ 1, 1024 ], "int32" ], "query_mask": [ [ 1, 32 ], "int32" ], "start_logits": [ [ 1, 32, 1025 ], "float16" ], "end_logits": [ [ 1, 32, 1025 ], "float16" ], "inside_logits": [ [ 1, 32, 1024 ], "float16" ], "spans": [ [ 1, 64, 2 ], "int32" ], "span_mask": [ [ 1, 64 ], "int32" ] }, "outputs": { "explicit_logits": [ [ 1, 32, 64 ], "float16" ] } } }