Searched refs:entry_computation_layout_ (Results 1 – 4 of 4) sorted by relevance
50 return entry_computation_layout_.has_value(); in has_entry_computation_layout()60 CHECK(entry_computation_layout_.has_value()); in entry_computation_layout()61 return *entry_computation_layout_; in entry_computation_layout()67 CHECK(entry_computation_layout_.has_value()); in mutable_entry_computation_layout()68 return &(*entry_computation_layout_); in mutable_entry_computation_layout()122 absl::optional<ComputationLayout> entry_computation_layout_;
33 : entry_computation_layout_( in HloModuleConfig()38 entry_computation_layout_ = ComputationLayout(program_shape); in SetDefaultComputationLayout()46 entry_computation_layout_->parameter_layouts()) { in compilation_cache_key()50 entry_computation_layout_->result_shape().SerializeAsString()); in compilation_cache_key()
972 : entry_computation_layout_(entry_computation_layout), in LayoutAssignment()984 << entry_computation_layout_->ToString(); in LayoutAssignment()1718 computation_layout == entry_computation_layout_); in RunOnComputation()1923 TF_RET_CHECK(entry_computation_layout_->parameter_count() == in Run()1927 ShapeUtil::Compatible(entry_computation_layout_->parameter_shape(i), in Run()1930 TF_RET_CHECK(ShapeUtil::Compatible(entry_computation_layout_->result_shape(), in Run()1958 entry_computation_layout_, *points_to_analysis, in Run()1970 entry_computation_layout_)); in Run()2093 *entry_computation_layout_ = saved_entry_computation_layout_; in Init()
443 ComputationLayout* entry_computation_layout_; variable