Searched refs:entry_layout (Results 1 – 1 of 1) sorted by relevance
380 ComputationLayout* entry_layout = in CreateModuleConfigFromProto() local382 for (int64 i = 0; i < entry_layout->parameter_count(); ++i) { in CreateModuleConfigFromProto()384 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromProto()387 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromProto()