Searched refs:CalculateComputationLayout (Results 1 – 2 of 2) sorted by relevance
426 Status CalculateComputationLayout(HloComputation* computation);
1665 Status LayoutAssignment::CalculateComputationLayout( in CalculateComputationLayout() function in xla::LayoutAssignment1784 TF_RETURN_IF_ERROR(CalculateComputationLayout(computation)); in RunOnComputation()