Searched refs:is_infeed (Results 1 – 1 of 1) sorted by relevance
135 absl::optional<Shape> GetXfeedShape(bool is_infeed, in GetXfeedShape() argument141 if (instruction.opcode() == HloOpcodeString(is_infeed in GetXfeedShape()168 string xfeed_name = is_infeed ? "infeed" : "outfeed"; in GetXfeedShape()170 is_infeed ? opts.fake_infeed_shape : opts.fake_outfeed_shape; in GetXfeedShape()172 is_infeed ? opts.generate_fake_infeed : opts.generate_fake_outfeed; in GetXfeedShape()186 xfeed_shape = is_infeed in GetXfeedShape()