[Debug + Feature]
Feature: input tensor shape inference passing
output_edge_map to allow ops absorbed by TRT subgraph to infer shape without
running another shape infer
Debug: fixed BiasAdd broadcasting
Debug: fixed rewiring input edge to TRT_ENGINE_OP
TODO:
incoming edge check (shape / dimension)
TRT dimension requirement for 3.1 makes input tensor with 2 dimension (NC)
tricky to interpret.
Loading
Please sign in to comment