Searched refs:GetXfeedShape (Results 1 – 1 of 1) sorted by relevance
145 std::optional<Shape> GetXfeedShape(bool is_infeed, const HloModuleProto& module, in GetXfeedShape() function275 if (std::optional<Shape> infeed_shape = GetXfeedShape( in ReplayComputation()281 GetXfeedShape(/*is_infeed=*/false, computation.proto(), opts); in ReplayComputation()