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