def METHOD_NAME(self) -> None:
    net = BasicModel_MultiLayer(inplace=True)
    inp = torch.tensor([[0.0, 5.0, 4.0]])
    self._gradient_input_test_assert(
        net, net.relu, inp, (0,), [[1.0, 1.0, 1.0]], attribute_to_neuron_input=True
    )