Searched refs:mutable_entry_computation_layout (Results 1 – 14 of 14) sorted by relevance
268 module->mutable_entry_computation_layout()->parameter_count()); in ForceParameterLayout()269 module->mutable_entry_computation_layout() in ForceParameterLayout()277 module->mutable_entry_computation_layout() in ForceResultLayout()284 module->mutable_entry_computation_layout() in ForceResultLayout()292 module->mutable_entry_computation_layout() in ForceClearResultLayout()
144 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function145 return config_.mutable_entry_computation_layout(); in mutable_entry_computation_layout()
51 config->mutable_entry_computation_layout(); in CreateModuleConfig()
102 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function
1300 m->mutable_entry_computation_layout() in AssignLayoutsToComputation()1305 m->mutable_entry_computation_layout(), in AssignLayoutsToComputation()1334 *m->mutable_entry_computation_layout()->mutable_result_layout() = in TEST_F()
137 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()157 *config.mutable_entry_computation_layout()->mutable_parameter_layout( in ReplaceEntryComputationParameter()
471 module_config.mutable_entry_computation_layout(); in CreateModuleConfigFromShape()
384 new_config.mutable_entry_computation_layout() in PadAndExecute()
858 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()863 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()
90 hlo_module->mutable_entry_computation_layout(), in RunHloOptimization()
66 config->mutable_entry_computation_layout(); in CreateModuleConfig()
393 module->mutable_entry_computation_layout(), in RunHloPassesThroughLayoutAssn()
300 hlo_module->mutable_entry_computation_layout(), in OptimizeHloModule()
3711 *config.mutable_entry_computation_layout() = in Run()