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