// Calculate min/max statistics.
    const auto input_flat = input_tensor.flat<float>();
    Tensor *min_output = nullptr, *max_output = nullptr;
    OP_REQUIRES_OK(context, context->allocate_output("min", {}, &min_output));
    OP_REQUIRES_OK(context, context->allocate_output("max", {}, &max_output));
    min_output->scalar<float>().device(
        context->template eigen_device<CPUDevice>()) = input_flat.minimum();
    max_output->scalar<float>().device(