Searched defs:computations (Results 1 – 14 of 14) sorted by relevance
196 absl::Span<HloComputation* const> computations) { in RootInstructions()318 absl::Span<HloComputation* const> computations) { in VerifyComputations()349 absl::Span<HloComputation* const> computations) { in ComputeReachability()
26 const auto& computations = module->computations(); in Run() local
152 computations() const { in computations() function158 computations() { in computations() function
35 std::vector<HloComputation*> computations = in Run() local
66 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime()
52 absl::Span<HloComputation* const> computations) { in CreateCallComputation()
307 std::vector<std::unique_ptr<HloComputation>> computations; in CreateFromProto() local
330 repeated HloComputationProto computations = 3; field
986 string ComputationsToString(absl::Span<HloComputation* const> computations) { in ComputationsToString()
110 const auto computations = [&computation_map, &proto](int index) { in CreateFromProto() local
26 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime()
329 absl::Span<const XlaComputationInstance> computations) { in ExecuteParallel()
464 std::vector<XlaComputation> computations; in TEST_F() local
499 auto computations = module->computations(); in FindComputation() local