Searched refs:entry_computation_ (Results 1 – 2 of 2) sorted by relevance
55 CHECK_EQ(nullptr, entry_computation_); in AddComputationInternal()56 entry_computation_ = computation.get(); in AddComputationInternal()62 entry_computation_->ComputeProgramShape()); in AddComputationInternal()65 entry_computation_->root_instruction()->shape()); in AddComputationInternal()194 entry_computation_ = tensorflow::gtl::FindWithDefault( in ReplaceComputations()195 replacements, entry_computation_, entry_computation_); in ReplaceComputations()227 proto.set_entry_computation_name(entry_computation_->name()); in ToProto()228 proto.set_entry_computation_id(entry_computation_->unique_id()); in ToProto()342 TF_RET_CHECK(module->entry_computation_ != nullptr); in CreateFromProto()570 auto cloned_computation = entry_computation_->Clone(suffix, &context); in Clone()
108 CHECK_NE(nullptr, entry_computation_); in entry_computation()109 return entry_computation_; in entry_computation()116 CHECK_NE(nullptr, entry_computation_); in result_shape()293 HloComputation* entry_computation_ = nullptr; variable