Searched defs:branch_computations (Results 1 – 4 of 4) sorted by relevance
1394 absl::Span<HloComputation* const> branch_computations, in CreateConditional()2711 const std::vector<HloComputation*>& HloInstruction::branch_computations() in branch_computations() function in xla::HloInstruction
2899 absl::Span<const ProgramShape> branch_computations, in InferConditionalShape()
2314 optional<std::vector<HloComputation*>> branch_computations; in ParseInstructionRhs() local
2391 absl::Span<const XlaComputation* const> branch_computations, in Conditional()2407 absl::Span<const XlaComputation* const> branch_computations, in ConditionalImpl()4725 absl::Span<const XlaComputation* const> branch_computations, in Conditional()