Searched refs:entry_layout (Results 1 – 1 of 1) sorted by relevance
571 ComputationLayout* entry_layout = in CreateModuleConfigFromShape() local573 for (int64_t i = 0; i < entry_layout->parameter_count(); ++i) { in CreateModuleConfigFromShape()575 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromShape()578 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromShape()