Searched refs:mutable_entry_computation_layout (Results 1 – 11 of 11) sorted by relevance
267 module->mutable_entry_computation_layout()->parameter_count()); in ForceParameterLayout()268 module->mutable_entry_computation_layout() in ForceParameterLayout()276 module->mutable_entry_computation_layout() in ForceResultLayout()283 module->mutable_entry_computation_layout() in ForceResultLayout()291 module->mutable_entry_computation_layout() in ForceClearResultLayout()
127 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function128 return config_.mutable_entry_computation_layout(); in mutable_entry_computation_layout()
66 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout() function
1306 m->mutable_entry_computation_layout() in AssignLayoutsToComputation()1311 m->mutable_entry_computation_layout(), in AssignLayoutsToComputation()1340 *m->mutable_entry_computation_layout()->mutable_result_layout() = in TEST_F()
381 module_config.mutable_entry_computation_layout(); in CreateModuleConfigFromProto()
134 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()
270 config->mutable_entry_computation_layout(); in CreateModuleConfig()
535 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()540 TF_CHECK_OK(module->mutable_entry_computation_layout() in ParseComputations()
87 hlo_module->mutable_entry_computation_layout(), in RunHloOptimization()
299 hlo_module->mutable_entry_computation_layout(), in OptimizeHloModule()
324 module->mutable_entry_computation_layout(), in RunHloPassesThroughLayoutAssn()