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