Searched defs:outfeed_shape (Results 1 – 9 of 9) sorted by relevance
68 Shape outfeed_shape; member
142 xla.ShapeProto outfeed_shape = 29; field
1197 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
405 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local1116 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed()4111 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
2107 optional<Shape> outfeed_shape; in ParseInstructionRhs() local
2087 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction()
92 auto outfeed_shape = ShapeUtil::MakeShape(F32, {4, 8}); in TEST_F() local
276 if (absl::optional<Shape> outfeed_shape = GetXfeedShape( in ReplayComputation() local
2912 Shape outfeed_shape = operand->shape(); in HandleOutfeed() local