crypten/gradients.py [1286:1302]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    @staticmethod
    def forward(ctx, *args, **kwargs):

        # preprocess inputs:
        assert len(args) >= 1
        if len(args) == 1:
            (input,) = args  # no dimension to min over in args
            dim = kwargs.pop("dim", None)  # remove dim from kwargs after obtaining it
        else:
            assert len(args) == 2
            assert "dim" not in kwargs
            input, dim = args  # dimension to min over in args
        keepdim = kwargs.get("keepdim", False)
        one_hot = kwargs.get("one_hot", True)

        # find minimum value (and corresponding argmin):
        if dim is None:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



crypten/gradients.py [1334:1349]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    @staticmethod
    def forward(ctx, *args, **kwargs):

        # preprocess inputs:
        assert len(args) >= 1
        if len(args) == 1:
            (input,) = args  # no dimension to max over in args
            dim = kwargs.pop("dim", None)  # remove dim from kwargs after obtaining it
        else:
            assert len(args) == 2
            assert "dim" not in kwargs
            input, dim = args  # dimension to max over in args
        keepdim = kwargs.get("keepdim", False)
        one_hot = kwargs.get("one_hot", True)
        # find maximum value (and corresponding argmax):
        if dim is None:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -