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