Searched refs:num_branches (Results 1 – 6 of 6) sorted by relevance
48 int num_branches = branches_.size(); in Compile() local49 OP_REQUIRES(ctx, num_branches >= 1, in Compile()93 num_branches); in Compile()94 std::vector<const FunctionBody*> case_bodies(num_branches); in Compile()95 for (int branch_idx = 0; branch_idx < num_branches; branch_idx++) { in Compile()110 for (int branch_idx = 0; branch_idx < num_branches; branch_idx++) { in Compile()132 std::vector<XlaCompiler::CompilationResult> branch_results(num_branches); in Compile()133 std::vector<XlaCompiler::CompilationResult*> branch_results_p(num_branches); in Compile()134 for (int j = 0; j < num_branches; ++j) { in Compile()171 for (int j = 0; j < num_branches; ++j) { in Compile()[all …]
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 …]
279 int num_branches = branches_attr->list().func_size(); in RewriteCaseNode() local281 branch_fn_names.reserve(num_branches); in RewriteCaseNode()282 for (int b = 0; b < num_branches; b++) { in RewriteCaseNode()
815 const int num_branches = conditional->branch_count(); in HandleConditional() local818 TF_RET_CHECK(num_branches == 2); in HandleConditional()827 TF_RET_CHECK(num_branches >= 1); in HandleConditional()829 TF_RETURN_IF_ERROR(CheckOperandCount(conditional, num_branches + 1)); in HandleConditional()830 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()
2778 int num_branches = conditional->branch_count(); in HandleConditional() local2785 for (int b = 0; b < num_branches; ++b) { in HandleConditional()2859 EmitGlobalCall(*conditional->branch_computation(num_branches - 1), in HandleConditional()2866 b_.CreateSwitch(branch_index_value, default_block, num_branches - 1); in HandleConditional()2868 for (int b = 0; b < num_branches - 1; ++b) { // last branch is default in HandleConditional()