Searched refs:mutable_outfeed_shape (Results 1 – 5 of 5) sorted by relevance
2026 Shape* mutable_outfeed_shape();
1561 Shape* mutable_outfeed_shape() { return &outfeed_shape_; } in mutable_outfeed_shape() function
4688 Shape* HloInstruction::mutable_outfeed_shape() { in mutable_outfeed_shape() function in xla::HloInstruction4689 return Cast<HloOutfeedInstruction>(this)->mutable_outfeed_shape(); in mutable_outfeed_shape()
2399 *proto.mutable_outfeed_shape() = outfeed_shape().ToProto(); in ToProto()
1872 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in Outfeed()1931 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in OutfeedWithTokenInternal()