Searched defs:branch_operands (Results 1 – 3 of 3) sorted by relevance
2558 absl::Span<const Shape> branch_operands) { in InferConditionalShape()
1904 absl::Span<const XlaOp> branch_operands) { in Conditional()3413 absl::Span<const XlaOp> branch_operands) { in Conditional()
2158 std::vector<BufferAllocation::Slice> branch_operands; in BuildConditionalThunk() local