Searched refs:LayoutShapeForChannel (Results 1 – 2 of 2) sorted by relevance
251 Shape LayoutShapeForChannel(Shape shape, int64 channel_id) const { in LayoutShapeForChannel() function
491 ->LayoutShapeForChannel(send_buffer_shape, in AddMandatoryConstraints()504 ->LayoutShapeForChannel( in AddMandatoryConstraints()522 ->LayoutShapeForChannel(buffer_shape, all_reduce_id); in AddMandatoryConstraints()