Lines Matching defs:HloComputation
52 // An HloComputation contains a directed acyclic graph of HLO instructions. The
55 class HloComputation {
57 // Builder class for HloComputation.
66 // Build and return an HloComputation. The parameter root_instruction
70 std::unique_ptr<HloComputation> Build(
163 // computation_map: a map from computation name to HloComputation*. This map
171 static StatusOr<std::unique_ptr<HloComputation>> CreateFromProto(
173 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
174 const std::function<void(std::unique_ptr<HloComputation>)>&
222 std::list<HloComputation*> MakeEmbeddedComputationsList() const;
255 bool operator==(const HloComputation& other) const;
305 std::unique_ptr<HloComputation> Clone(const string& suffix = "clone",
313 std::unique_ptr<HloComputation> CloneWithReplacements(
346 explicit HloComputation(
395 TF_DISALLOW_COPY_AND_ASSIGN(HloComputation);
