ComparisonDirection direction,
                                        Region *body, OpBuilder *builder) {
  OpBuilder::InsertionGuard insertion_point_gurad(*builder);

  Type input_type =
      tensorflow::GetTypeFromTFTensorShape(/*shape=*/{}, input_element_type);
  Type index_type =
      tensorflow::GetTypeFromTFTensorShape(/*shape=*/{}, index_element_type);