Searched refs:PropagateComputationLayouts (Results 1 – 2 of 2) sorted by relevance
439 Status PropagateComputationLayouts(HloComputation* computation,
1889 Status LayoutAssignment::PropagateComputationLayouts( in PropagateComputationLayouts() function in xla::LayoutAssignment1969 TF_RETURN_IF_ERROR(PropagateComputationLayouts(module->entry_computation(), in Run()