Searched refs:num_branches (Results 1 – 5 of 5) sorted by relevance
193 int num_branches = GetParam(); in XLA_TEST_P() local194 for (int bi = -1; bi <= num_branches; ++bi) { in XLA_TEST_P()202 std::vector<XlaOp> operands(num_branches, operand); in XLA_TEST_P()204 branches.reserve(num_branches); in XLA_TEST_P()205 std::vector<const XlaComputation*> branches_p(num_branches); in XLA_TEST_P()206 for (int i = 0; i < num_branches; ++i) { in XLA_TEST_P()213 float expected = 10 * static_cast<float>((bi < 0 || bi >= num_branches) in XLA_TEST_P()214 ? num_branches - 1 in XLA_TEST_P()236 int num_branches = GetParam(); in XLA_TEST_P() local237 for (int bi = -1; bi <= num_branches; ++bi) { in XLA_TEST_P()[all …]
43 int num_branches = branches_.size(); in Compile() local44 OP_REQUIRES(ctx, num_branches >= 1, in Compile()102 std::vector<XlaCompiler::CompilationResult> branch_results(num_branches); in Compile()103 std::vector<XlaCompiler::CompilationResult*> branch_results_p(num_branches); in Compile()104 for (int j = 0; j < num_branches; ++j) { in Compile()141 for (int j = 0; j < num_branches; ++j) { in Compile()150 std::vector<const xla::XlaComputation*> result_computations(num_branches); in Compile()151 for (int j = 0; j < num_branches; ++j) { in Compile()237 std::vector<xla::XlaOp>(num_branches, input_tuple)); in Compile()
670 const int num_branches = conditional->branch_count(); in HandleConditional() local672 TF_RET_CHECK(num_branches == 2); in HandleConditional()674 TF_RET_CHECK(num_branches >= 1); in HandleConditional()676 TF_RETURN_IF_ERROR(CheckOperandCount(conditional, num_branches + 1)); in HandleConditional()677 for (int j = 0; j < num_branches; ++j) { in HandleConditional()
655 int num_branches, in AddNBranches() argument660 for (int i = 0; i < num_branches; ++i) { in AddNBranches()
2533 int num_branches = conditional->branch_count(); in HandleConditional() local2540 for (int b = 0; b < num_branches; ++b) { in HandleConditional()2614 EmitGlobalCall(*conditional->branch_computation(num_branches - 1), in HandleConditional()2621 b_.CreateSwitch(branch_index_value, default_block, num_branches - 1); in HandleConditional()2623 for (int b = 0; b < num_branches - 1; ++b) { // last branch is default in HandleConditional()