HomeSort by relevance Sort by last modified time
    Searched defs:HloComputation (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation.h 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
    [all...]
hlo_computation.cc 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)
    [all...]

Completed in 90 milliseconds