e086b02bf4d19fbdcc751bfa876d03f2fcdaf375,tf_encrypted/tensor/prime.py,PrimeFactory,constant,#PrimeFactory#Any#,250

Before Change

    def constant(self, value) -> PrimeConstant:

        if isinstance(value, (tf.Tensor, np.ndarray)):
            return PrimeConstant(value, self)

        if isinstance(value, PrimeTensor):
            assert value.modulus == self.modulus, \
                "Incompatible modulus: {}, (expected {})".format(value.modulus, self.modulus)
            return PrimeConstant(value.value, self)

        raise TypeError("Don"t know how to handle {}".format(type(value)))

    def variable(self, initial_value) -> PrimeVariable:

After Change

    def constant(self, value) -> PrimeConstant:

        if isinstance(value, np.ndarray):
            constant = tf.constant(value, dtype=self.native_type)
            return PrimeConstant(constant, self)

        raise TypeError("Don"t know how to handle {}".format(type(value)))

Instances