Searched defs:outfeed_shape (Results 1 – 9 of 9) sorted by relevance
68 Shape outfeed_shape; member
197 xla.ShapeProto outfeed_shape = 29; field
462 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local1390 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed()4684 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
1559 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
2457 optional<Shape> outfeed_shape; in CreateInstruction() local
2385 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction()
102 auto outfeed_shape = ShapeUtil::MakeShape(F32, {4, 8}); in TEST_F() local
280 std::optional<Shape> outfeed_shape = in ReplayComputation() local
3693 Shape outfeed_shape = operand->shape(); in HandleOutfeed() local