Searched refs:new_program_shape (Results 1 – 1 of 1) sorted by relevance
4521 auto new_program_shape = module->entry_computation()->ComputeProgramShape(); in Run() local4524 program_shape.result(), new_program_shape.result())) in Run()4527 new_program_shape.parameters_size()) in Run()4531 program_shape.parameters(i), new_program_shape.parameters(i))) in Run()4543 for (int64_t i = 0; i < new_program_shape.parameters_size(); ++i) { in Run()4546 new_program_shape.mutable_parameters(i))); in Run()4547 ShapeUtil::ForEachMutableSubshape(new_program_shape.mutable_parameters(i), in Run()4551 old_entry_layout.result_shape(), new_program_shape.mutable_result())); in Run()4552 ShapeUtil::ForEachMutableSubshape(new_program_shape.mutable_result(), in Run()4557 ComputationLayout(new_program_shape, /*ignore_layouts=*/false); in Run()