Searched defs:branch_operands (Results 1 – 5 of 5) sorted by relevance
2786 absl::Span<const Shape> branch_operands) { in InferConditionalShape()
658 llvm::SmallVector<xla::XlaOp, 4> branch_operands(branches.size()); in ExportXlaOp() local
2364 absl::Span<const XlaOp> branch_operands) { in Conditional()2380 absl::Span<const XlaOp> branch_operands) { in ConditionalImpl()4864 absl::Span<const XlaOp> branch_operands) { in Conditional()
2103 OperandRange branch_operands = op.branch_operands(); in Verify() local
3900 std::vector<BufferAllocation::Slice> branch_operands; in BuildConditionalThunk() local