Lines Matching defs:HloComputation
47 std::unique_ptr<HloComputation> HloComputation::Builder::Build(
59 return WrapUnique(new HloComputation(name_, parameter_count, &instructions_,
63 HloComputation::HloComputation(
90 HloInstruction* HloComputation::AddInstruction(
98 HloInstruction* HloComputation::AddInstructionInternal(
111 HloInstruction* HloComputation::AddParameter(
122 Status HloComputation::RemoveParameter(int64 param_no) {
161 void HloComputation::Reparent(HloInstruction* instruction) {
165 bool HloComputation::IsRemovable(const HloInstruction* instruction) {
182 bool HloComputation::HasSideEffect() const {
191 Status HloComputation::RemoveInstructionAndUnusedOperands(
220 Status HloComputation::RemoveInstruction(HloInstruction* instruction) {
245 void HloComputation::set_root_instruction(
307 HloComputation* computation,
308 tensorflow::gtl::FlatSet<HloComputation*>* visited,
309 std::list<HloComputation*>* post_order) {
315 for (HloComputation* called_computation :
327 std::list<HloInstruction*> HloComputation::MakeInstructionPostOrder() const {
349 std::list<HloComputation*> HloComputation::MakeEmbeddedComputationsList()
351 tensorflow::gtl::FlatSet<HloComputation*> visited;
352 std::list<HloComputation*> post_order;
357 ComputeComputationPostOrder(const_cast<HloComputation*>(this), &visited,
367 string HloComputation::ToString(const HloPrintOptions& options) const {
394 HloComputationProto HloComputation::ToProto() const {
405 /* static */ StatusOr<std::unique_ptr<HloComputation>>
406 HloComputation::CreateFromProto(
408 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
409 const std::function<void(std::unique_ptr<HloComputation>)>&
431 return WrapUnique(new HloComputation(
435 void HloComputation::FuseInstructionsInto(
454 HloInstruction* HloComputation::CreateFusionInstruction(
464 StatusOr<HloInstruction*> HloComputation::DeepCopyHelper(
504 StatusOr<HloInstruction*> HloComputation::DeepCopyInstruction(
526 ProgramShape HloComputation::ComputeProgramShape() const {
539 bool HloComputation::operator==(const HloComputation& other) const {
551 *b, eq, [](const HloComputation* a, const HloComputation* b) {
558 Status HloComputation::ReplaceWithNewInstruction(
565 Status HloComputation::ReplaceInstruction(HloInstruction* old_instruction,
587 std::unique_ptr<HloReachabilityMap> HloComputation::ComputeReachability()
602 void HloComputation::UpdateReachabilityThroughInstruction(
629 std::vector<HloInstruction*> HloComputation::CollectUnreachableRoots() const {
648 Status HloComputation::Accept(
663 template Status HloComputation::Accept(DfsHloVisitor* visitor) const;
664 template Status HloComputation::Accept(ConstDfsHloVisitor* visitor) const;
666 Status HloComputation::AcceptWithOperandOrder(
683 Status HloComputation::AcceptOrdered(
714 template Status HloComputation::AcceptOrdered(
716 template Status HloComputation::AcceptOrdered(
719 Status HloComputation::Accept(
725 Status HloComputation::Accept(
731 std::unique_ptr<HloComputation> HloComputation::Clone(const string& suffix,
739 std::unique_ptr<HloComputation> HloComputation::CloneWithReplacements(
821 void HloComputation::UniquifyName(NameUniquer* name_uniquer) {
