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 // You can think of an HloComputation like a function. It has some inputs
66 class HloComputation {
68 // Builder class for HloComputation.
77 // Build and return an HloComputation. The parameter root_instruction
81 std::unique_ptr<HloComputation> Build(
194 // computation_map: a map from computation id to HloComputation*. This map
197 static StatusOr<std::unique_ptr<HloComputation>> CreateFromProto(
199 const absl::flat_hash_map<int64, HloComputation*>& computation_map);
232 std::vector<HloComputation*> MakeEmbeddedComputationsList() const;
266 HloComputation* computation)>& copy_leaf)
    [all...]
hlo_computation.cc 51 std::unique_ptr<HloComputation> HloComputation::Builder::Build(
63 return absl::WrapUnique(new HloComputation(
67 HloComputation::HloComputation(
95 HloInstruction* HloComputation::AddInstruction(
103 HloInstruction* HloComputation::AddInstructionInternal(
116 HloInstruction* HloComputation::AddParameter(
127 HloInstruction* HloComputation::AddEntryComputationParameter(
145 Status HloComputation::RemoveParameter(int64 param_no)
    [all...]

Completed in 54 milliseconds