Searched refs:branch_results (Results 1 – 2 of 2) sorted by relevance
162 std::vector<XlaCompiler::CompilationResult> branch_results(num_branches); in Compile() local166 &branch_results[j])); in Compile()170 for (XlaCompiler::CompilationResult& result : branch_results) { in Compile()200 branch_results[j] = {}; in Compile()203 &branch_results[j])); in Compile()211 OP_REQUIRES(ctx, branch_results[j].xla_input_shapes.size() == 1, in Compile()213 xla::Shape branch_input_shape = branch_results[j].xla_input_shapes[0]; in Compile()230 xla::ShapeUtil::Compatible(branch_results[0].xla_output_shape, in Compile()231 branch_results[j].xla_output_shape), in Compile()234 xla::ShapeUtil::HumanString(branch_results[0].xla_output_shape), in Compile()[all …]
850 TypeRangeWithDesc branch_results{yield->getOperandTypes(), description}; in Verify() local851 if (failed(VerifyTypeRangesAreCompatible(op, branch_results, results))) in Verify()