Searched refs:mutable_entry_computation_layout (Results 1 – 14 of 14) sorted by relevance
303 module->mutable_entry_computation_layout()->parameter_count()); in ForceParameterLayout()304 module->mutable_entry_computation_layout() in ForceParameterLayout()312 module->mutable_entry_computation_layout() in ForceResultLayout()319 module->mutable_entry_computation_layout() in ForceResultLayout()327 module->mutable_entry_computation_layout() in ForceClearResultLayout()
51 config->mutable_entry_computation_layout(); in CreateModuleConfig()172 *module->mutable_entry_computation_layout()->mutable_parameter_layout(i) = in UpdateEntryComputationLayout()177 *module->mutable_entry_computation_layout()->mutable_result_layout() = in UpdateEntryComputationLayout()
148 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function149 return config_.mutable_entry_computation_layout(); in mutable_entry_computation_layout()
106 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function
1336 m->mutable_entry_computation_layout() in AssignLayoutsToComputation()1340 LayoutAssignment layout_assignment(m->mutable_entry_computation_layout(), in AssignLayoutsToComputation()1369 *m->mutable_entry_computation_layout()->mutable_result_layout() = in TEST_F()
171 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()191 *config.mutable_entry_computation_layout()->mutable_parameter_layout( in ReplaceEntryComputationParameter()
572 module_config.mutable_entry_computation_layout(); in CreateModuleConfigFromShape()
539 new_config.mutable_entry_computation_layout() in PadAndExecute()
954 *config.mutable_entry_computation_layout() = *entry_computation_layout; in ParseHloModule()1001 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()1006 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()
96 hlo_module->mutable_entry_computation_layout()); in RunHloOptimization()
68 config->mutable_entry_computation_layout(); in CreateModuleConfig()
613 module->mutable_entry_computation_layout(), target_machine_features, in RunHloPassesThroughLayoutAssn()
599 hlo_module->mutable_entry_computation_layout(), stream_exec, in OptimizeHloModule()
4556 *config.mutable_entry_computation_layout() = in Run()