Searched refs:outside_shape (Results 1 – 3 of 3) sorted by relevance
454 conn.outside_shape.AsProto(&in_shape); in CreateTRTNode()460 input_shapes.at(conn.port_number) = conn.outside_shape; in CreateTRTNode()464 for (int i = 1; i < conn.outside_shape.dims(); i++) { in CreateTRTNode()465 if (conn.outside_shape.dim_size(i) <= 0) { in CreateTRTNode()469 conn.outside_shape.DebugString(), ")"); in CreateTRTNode()
83 PartialTensorShape outside_shape; // Only set for input edge. member
4192 connection.outside_shape = partial_shape; in ConvertSegmentToGraphDef()