Searched defs:ComputeProgramShape (Results 1 – 2 of 2) sorted by relevance
74 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape() function in xla::ComputationLayout
878 ProgramShape HloComputation::ComputeProgramShape(bool include_ids) const { in ComputeProgramShape() function in xla::HloComputation