Searched refs:new_program_shape (Results 1 – 1 of 1) sorted by relevance
3686 auto new_program_shape = module->entry_computation()->ComputeProgramShape(); in Run() local3689 program_shape.result(), new_program_shape.result())) in Run()3692 new_program_shape.parameters_size()) in Run()3696 program_shape.parameters(i), new_program_shape.parameters(i))) in Run()3702 for (int64 i = 0; i < new_program_shape.parameters_size(); ++i) { in Run()3705 new_program_shape.mutable_parameters(i))); in Run()3708 old_entry_layout.result_shape(), new_program_shape.mutable_result())); in Run()3712 ComputationLayout(new_program_shape, /*ignore_layouts=*/false); in Run()