Searched defs:outfeed_shape (Results 1 – 6 of 6) sorted by relevance
71 Shape outfeed_shape; member
138 xla.ShapeProto outfeed_shape = 29; field
1005 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
368 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local870 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed()3404 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
1812 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction()
244 if (absl::optional<Shape> outfeed_shape = GetXfeedShape( in ReplayComputation() local