Searched refs:branch_input_shape (Results 1 – 1 of 1) sorted by relevance
213 xla::Shape branch_input_shape = branch_results[j].xla_input_shapes[0]; in Compile() local215 branch0_input_shape = branch_input_shape; in Compile()217 OP_REQUIRES(ctx, branch_input_shape.IsTuple(), in Compile()221 xla::ShapeUtil::Compatible(branch0_input_shape, branch_input_shape), in Compile()225 xla::ShapeUtil::HumanString(branch_input_shape))); in Compile()