Searched defs:branch_computations (Results 1 – 7 of 7) sorted by relevance
168 absl::Span<const XlaComputation* const> branch_computations, in DynamicConditional()
290 const auto& branch_computations = instruction->branch_computations(); in GetAllTuples() local
1491 absl::Span<HloComputation* const> branch_computations, in CreateConditional()2872 const std::vector<HloComputation*>& HloInstruction::branch_computations() in branch_computations() function in xla::HloInstruction
1210 const auto& branch_computations = instruction->branch_computations(); in CheckLayouts() local
2902 absl::Span<const ProgramShape> branch_computations, in InferConditionalShape()
2525 optional<std::vector<HloComputation*>> branch_computations; in CreateInstruction() local
2518 absl::Span<const XlaComputation* const> branch_computations, in Conditional()2534 absl::Span<const XlaComputation* const> branch_computations, in ConditionalImpl()4887 absl::Span<const XlaComputation* const> branch_computations, in Conditional()