Searched refs:input_shape_protos (Results 1 – 1 of 1) sorted by relevance
336 std::vector<tensorflow::TensorShapeProto> input_shape_protos; in CreateTRTNode() local378 input_shape_protos.resize(conn.port_number + 1); in CreateTRTNode()381 conn.outside_shape.AsProto(&input_shape_protos.at(conn.port_number)); in CreateTRTNode()485 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()