Searched defs:branch_computations (Results 1 – 4 of 4) sorted by relevance
1217 absl::Span<HloComputation* const> branch_computations, in CreateConditional()2511 const std::vector<HloComputation*>& HloInstruction::branch_computations() in branch_computations() function in xla::HloInstruction
2785 absl::Span<const ProgramShape> branch_computations, in InferConditionalShape()
2177 optional<std::vector<HloComputation*>> branch_computations; in ParseInstructionRhs() local
2363 absl::Span<const XlaComputation* const> branch_computations, in Conditional()2379 absl::Span<const XlaComputation* const> branch_computations, in ConditionalImpl()4863 absl::Span<const XlaComputation* const> branch_computations, in Conditional()