Commit 35932989 authored by Jie's avatar Jie
Browse files

[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.
parent 51ce6cf0
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment