Searched refs:entry_computation_layout_ (Results 1 – 4 of 4) sorted by relevance
33 : entry_computation_layout_( in HloModuleConfig()37 : entry_computation_layout_(std::move(entry_computation_layout)) {} in HloModuleConfig()41 entry_computation_layout_ = ComputationLayout(program_shape); in SetDefaultComputationLayout()46 entry_computation_layout_ = ComputationLayout(program_shape, in SetComputationLayoutIfExists()54 if (entry_computation_layout_.has_value()) { in compilation_cache_key()56 entry_computation_layout_->parameter_layouts()) { in compilation_cache_key()60 entry_computation_layout_->result_shape().SerializeAsString()); in compilation_cache_key()
85 return entry_computation_layout_.has_value(); in has_entry_computation_layout()100 CHECK(entry_computation_layout_.has_value()); in entry_computation_layout()101 return *entry_computation_layout_; in entry_computation_layout()107 CHECK(entry_computation_layout_.has_value()); in mutable_entry_computation_layout()108 return &(*entry_computation_layout_); in mutable_entry_computation_layout()113 entry_computation_layout_ = std::nullopt; in clear_entry_computation_layout()352 std::optional<ComputationLayout> entry_computation_layout_;
605 entry_computation_layout_->LayoutIsSet()) || in AddMandatoryConstraints()803 entry_computation_layout_->LayoutIsSet()) || in AddMandatoryConstraints()1244 : entry_computation_layout_(entry_computation_layout), in LayoutAssignment()1254 << entry_computation_layout_->ToString(); in LayoutAssignment()2196 *entry_computation_layout_ = constraints->computation_layout(); in CalculateComputationLayout()2460 TF_RET_CHECK(entry_computation_layout_->parameter_count() == in Run()2464 ShapeUtil::Compatible(entry_computation_layout_->parameter_shape(i), in Run()2467 TF_RET_CHECK(ShapeUtil::Compatible(entry_computation_layout_->result_shape(), in Run()2498 entry_computation_layout_->LayoutIsSet() in Run()2499 ? entry_computation_layout_ in Run()[all …]
552 ComputationLayout* entry_computation_layout_; variable