Searched defs:branch_operands (Results 1 – 5 of 5) sorted by relevance
169 absl::Span<const XlaOp> branch_operands) { in DynamicConditional()
2903 absl::Span<const Shape> branch_operands) { in InferConditionalShape()
1025 llvm::SmallVector<mlir::Value, 4> branch_operands; in matchAndRewrite() local
2519 absl::Span<const XlaOp> branch_operands) { in Conditional()2535 absl::Span<const XlaOp> branch_operands) { in ConditionalImpl()4888 absl::Span<const XlaOp> branch_operands) { in Conditional()
980 llvm::SmallVector<xla::XlaOp, 4> branch_operands(branches.size()); in ExportXlaOp() local