Searched refs:outside_shape (Results 1 – 3 of 3) sorted by relevance
381 conn.outside_shape.AsProto(&input_shape_protos.at(conn.port_number)); in CreateTRTNode()382 input_shapes.at(conn.port_number) = conn.outside_shape; in CreateTRTNode()387 for (int i = 1; i < conn.outside_shape.dims(); i++) { in CreateTRTNode()388 if (conn.outside_shape.dim_size(i) <= 0) { in CreateTRTNode()
77 PartialTensorShape outside_shape; // Only set for input edge. member
6478 connection.outside_shape = partial_shape; in ConvertSegmentToGraphDef()