conditional_test.cc | 197 XlaOp branch_index; local 199 &builder, &branch_index); 211 Conditional(branch_index, branches_p, operands); 240 XlaOp branch_index; local 242 &builder, &branch_index); 264 Conditional(branch_index, branches_p, operands); 415 XlaOp branch_index; local 417 CreateR0Parameter<int32>(bi, 0, "pred", &builder, &branch_index); 436 Conditional(branch_index, branches_p, [all...] |