Searched refs:output_shape_protos (Results 1 – 1 of 1) sorted by relevance
406 std::vector<TensorShapeProto> output_shape_protos; in CreateTRTNode() local445 if (output_shape_protos.size() <= conn.port_number) { in CreateTRTNode()446 output_shape_protos.resize(conn.port_number + 1); in CreateTRTNode()449 output_shape_protos.at(conn.port_number) = out_shape; in CreateTRTNode()556 .Attr("output_shapes", output_shape_protos) in CreateTRTNode()