Searched defs:branch_computations (Results 1 – 4 of 4) sorted by relevance
971 absl::Span<HloComputation* const> branch_computations, in CreateConditional()2069 const std::vector<HloComputation*>& HloInstruction::branch_computations() in branch_computations() function in xla::HloInstruction
2557 absl::Span<const ProgramShape> branch_computations, in InferConditionalShape()
1451 optional<std::vector<HloComputation*>> branch_computations; in ParseInstructionRhs() local
1903 absl::Span<const XlaComputation* const> branch_computations, in Conditional()3412 absl::Span<const XlaComputation* const> branch_computations, in Conditional()