Home
last modified time | relevance | path

Searched defs:outfeed_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.h71 Shape outfeed_shape; member
Dhlo.proto138 xla.ShapeProto outfeed_shape = 29; field
Dhlo_instructions.h1005 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
Dhlo_instruction.cc368 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local
870 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed()
3404 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
Dhlo_instructions.cc1812 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc244 if (absl::optional<Shape> outfeed_shape = GetXfeedShape( in ReplayComputation() local