Searched refs:entry_computation_ (Results 1 – 2 of 2) sorted by relevance
68 entry_computation_ = entry_computation; in ReplaceEntryComputation()70 entry_computation_->ComputeProgramShape()); in ReplaceEntryComputation()72 entry_computation_->root_instruction()->shape()); in ReplaceEntryComputation()79 CHECK_EQ(nullptr, entry_computation_); in AddComputationInternal()80 entry_computation_ = computation.get(); in AddComputationInternal()84 entry_computation_->ComputeProgramShape()); in AddComputationInternal()89 entry_computation_->ComputeProgramShape()); in AddComputationInternal()92 entry_computation_->root_instruction()->shape()); in AddComputationInternal()238 entry_computation_ = tensorflow::gtl::FindWithDefault( in ReplaceComputations()239 replacements, entry_computation_, entry_computation_); in ReplaceComputations()[all …]
127 CHECK_NE(nullptr, entry_computation_); in entry_computation()128 return entry_computation_; in entry_computation()131 bool has_entry_computation() const { return entry_computation_ != nullptr; } in has_entry_computation()137 CHECK_NE(nullptr, entry_computation_); in result_shape()530 HloComputation* entry_computation_ = nullptr; variable