Exemplos de _jax_type em Python

Exemplo n.º 1

Arquivo: dtypes_test.py Projeto: zhaowilliam/jax

 def testJaxTypeFromVal(self, jaxtype):
     try:
         val = jaxtype(0)
     except TypeError:
         val = jaxtype.type(0)
     self.assertIs(dtypes._jax_type(*dtypes._dtype_and_weaktype(val)),
                   jaxtype)

Exemplo n.º 2

 def testJaxTypeFromType(self, jaxtype):
     self.assertIs(dtypes._jax_type(jaxtype), jaxtype)

Exemplo n.º 3

Arquivo: dtypes_test.py Projeto: zhaowilliam/jax

 def testJaxTypeFromType(self, jaxtype):
     self.assertIs(dtypes._jax_type(*dtypes._dtype_and_weaktype(jaxtype)),
                   jaxtype)