Searched refs:input_shape_protos (Results 1 – 1 of 1) sorted by relevance
354 std::vector<tensorflow::TensorShapeProto> input_shape_protos; in CreateTRTNode() local410 /*shape_proto_vect=*/input_shape_protos, in CreateTRTNode()414 input_shape_protos.at(conn.port_number) = in_shape; in CreateTRTNode()491 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()