Searched defs:branch_computations (Results 1 – 4 of 4) sorted by relevance
1058 absl::Span<HloComputation* const> branch_computations, in CreateConditional()2210 const std::vector<HloComputation*>& HloInstruction::branch_computations() in branch_computations() function in xla::HloInstruction
2674 absl::Span<const ProgramShape> branch_computations, in InferConditionalShape()
1524 optional<std::vector<HloComputation*>> branch_computations; in ParseInstructionRhs() local
1854 absl::Span<const XlaComputation* const> branch_computations, in Conditional()1870 absl::Span<const XlaComputation* const> branch_computations, in ConditionalImpl()3478 absl::Span<const XlaComputation* const> branch_computations, in Conditional()