xla::HloComputation::Builder Class Reference

Builder class for HloComputation. More...

#include <hlo_computation.h>

Collaboration diagram for xla::HloComputation::Builder:

Public Member Functions
std::unique_ptr< HloComputation >	Build (HloInstruction *root_instruction=nullptr)
	Build and return an HloComputation. More...
HloInstruction *	AddInstruction (std::unique_ptr< HloInstruction > instruction)
	Add the instruction to the member instructions which is an array used to record instructions.

Detailed Description

Builder class for HloComputation.

Member Function Documentation

Build()

std::unique_ptr< HloComputation > xla::HloComputation::Builder::Build

(

HloInstruction *

root_instruction = nullptr

)

Build and return an HloComputation.

1. Traverse the instruction list to count the parameter_count.
2. Use the root instruction (can be specified otherwise use the last added instruction) to build a HloComputation object and return it.

• Implementation

---

The documentation for this class was generated from the following files:

• tensorflow/tensorflow/compiler/xla/service/hlo_computation.h
• tensorflow/tensorflow/compiler/xla/service/hlo_computation.cc