File size: 659 Bytes
948d40c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | name: "screen_sbert_engine"
platform: "tensorrt_plan"
input [
{
name: "bbox"
data_type: TYPE_FP32
dims: [ -1, -1, 4 ]
},
{
name: "function_embedding"
data_type: TYPE_FP32
dims: [ -1, -1, 1024 ]
},
{
name: "vision_embedding"
data_type: TYPE_FP32
dims: [ -1, -1, 768 ]
},
{
name: "text_embedding"
data_type: TYPE_FP32
dims: [ -1, -1, 1024 ]
},
{
name: "padding_mask"
data_type: TYPE_INT32
dims: [ -1, -1 ]
}
]
output [
{
name: "screen_embedding"
data_type: TYPE_FP32
dims: [ -1, 512 ]
}
]
instance_group [
{
kind: KIND_GPU
count: 1
gpus: [0]
}
]
|