Searched refs:entry_computation_ (Results 1 – 2 of 2) sorted by relevance
60 entry_computation_ = entry_computation; in ReplaceEntryComputation()62 entry_computation_->ComputeProgramShape()); in ReplaceEntryComputation()64 entry_computation_->root_instruction()->shape()); in ReplaceEntryComputation()71 CHECK_EQ(nullptr, entry_computation_); in AddComputationInternal()72 entry_computation_ = computation.get(); in AddComputationInternal()76 entry_computation_->ComputeProgramShape()); in AddComputationInternal()81 entry_computation_->ComputeProgramShape()); in AddComputationInternal()84 entry_computation_->root_instruction()->shape()); in AddComputationInternal()229 entry_computation_ = tensorflow::gtl::FindWithDefault( in ReplaceComputations()230 replacements, entry_computation_, entry_computation_); in ReplaceComputations()[all …]
123 CHECK_NE(nullptr, entry_computation_); in entry_computation()124 return entry_computation_; in entry_computation()127 bool has_entry_computation() const { return entry_computation_ != nullptr; } in has_entry_computation()133 CHECK_NE(nullptr, entry_computation_); in result_shape()389 HloComputation* entry_computation_ = nullptr; variable