Searched refs:input_shape_protos (Results 1 – 1 of 1) sorted by relevance
407 std::vector<TensorShapeProto> input_shape_protos; in CreateTRTNode() local455 if (input_shape_protos.size() <= conn.port_number) { in CreateTRTNode()456 input_shape_protos.resize(conn.port_number + 1); in CreateTRTNode()459 input_shape_protos.at(conn.port_number) = in_shape; in CreateTRTNode()555 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()