Searched defs:branch_operands (Results 1 – 4 of 4) sorted by relevance
990 llvm::SmallVector<mlir::Value, 4> branch_operands; in matchAndRewrite() local
2900 absl::Span<const Shape> branch_operands) { in InferConditionalShape()
750 llvm::SmallVector<xla::XlaOp, 4> branch_operands(branches.size()); in ExportXlaOp() local
2392 absl::Span<const XlaOp> branch_operands) { in Conditional()2408 absl::Span<const XlaOp> branch_operands) { in ConditionalImpl()4726 absl::Span<const XlaOp> branch_operands) { in Conditional()