Searched refs:entry_computation_ (Results 1 – 2 of 2) sorted by relevance
62 entry_computation_ = entry_computation; in ReplaceEntryComputation()64 entry_computation_->ComputeProgramShape()); in ReplaceEntryComputation()66 entry_computation_->root_instruction()->shape()); in ReplaceEntryComputation()73 CHECK_EQ(nullptr, entry_computation_); in AddComputationInternal()74 entry_computation_ = computation.get(); in AddComputationInternal()78 entry_computation_->ComputeProgramShape()); in AddComputationInternal()83 entry_computation_->ComputeProgramShape()); in AddComputationInternal()86 entry_computation_->root_instruction()->shape()); in AddComputationInternal()232 entry_computation_ = tensorflow::gtl::FindWithDefault( in ReplaceComputations()233 replacements, entry_computation_, entry_computation_); in ReplaceComputations()[all …]
124 CHECK_NE(nullptr, entry_computation_); in entry_computation()125 return entry_computation_; in entry_computation()128 bool has_entry_computation() const { return entry_computation_ != nullptr; } in has_entry_computation()134 CHECK_NE(nullptr, entry_computation_); in result_shape()394 HloComputation* entry_computation_ = nullptr; variable