Searched defs:send_shapes (Results 1 – 3 of 3) sorted by relevance
116 auto send_shapes = GetValue<ValuePtrList>(primitive->GetAttr(kSendShapes)); in Check() local
658 Attr send_shapes = {SEND_SHAPES, MakeTupleListValue(send_shapes_)}; in CreateNeighborExchangeAttrs() local
643 def __init__(self, send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, recv_type, argument