Exemplos de compile_tf_module em Python

Exemplo n.º 1

Arquivo: regression_training_test.py Projeto: sjw36/iree

 def __init__(self, *args, **kwargs):
   super().__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(
       RegressionTrainingModule,
       exported_names=["train_on_batch"],
       relative_artifacts_dir=os.path.join(RegressionTrainingModule.__name__,
                                           FLAGS.optimizer))

Exemplo n.º 2

  def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self._modules = tf_test_utils.compile_tf_module(FftModule)
    self.real_array = np.array([
        9., 1., 4.5, -0.3, 10., -1., 5.5, 0.3, 299., 3.5, -0.777, 2, 1.7, 3.5,
        -4.5, 0.0
    ],
                               dtype=np.float32)
    self.imag_array = np.array([
        0., -1., 17.7, 10., 0., -11., 2763, 0., 0., -1.5, 16.8, 100., 0., -111.,
        2.3, 1.
    ],
                               dtype=np.float32)

    # Required since pffft requires a minimum of 32 elements for real ffts.
    self.long_real_array = np.concatenate((self.real_array, self.real_array),
                                          axis=None)

Exemplo n.º 3

Arquivo: resource_ops_test.py Projeto: xinan-jiang/iree

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(ResourcesOpsModule)

Exemplo n.º 4

 def __init__(self, *args, **kwargs):
   super(DynamicMlpTest, self).__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(DynamicMlpModule,
                                                   exported_names=["predict"])

Exemplo n.º 5

Arquivo: slim_vision_model_test.py Projeto: xinan-jiang/iree

 def __init__(self, *args, **kwargs):
   super().__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(
       SlimVisionModule,
       exported_names=['predict'],
       relative_artifacts_dir=FLAGS.model)

Exemplo n.º 6

Arquivo: logical_ops_test.py Projeto: BernhardRiemann/iree

 def __init__(self, methodName="runTest"):
     super(LogicalOpsTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(LogicalOpsModule)

Exemplo n.º 7

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(DepthConv2dModule)

Exemplo n.º 8

 def __init__(self, *args, **kwargs):
   super(BatchNormTest, self).__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(BatchNormModule)

Exemplo n.º 9

 def __init__(self, *args, **kwargs):
   super(ControlFlowTest, self).__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(ControlFlowModule)

Exemplo n.º 10

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(BroadcastingModule)

Exemplo n.º 11

 def __init__(self, *args, **kwargs):
   super(SimpleArithmeticTest, self).__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(SimpleArithmeticModule)

Exemplo n.º 12

 def __init__(self, methodName="runTest"):
     super(BatchNormTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(BatchNormModule)

Exemplo n.º 13

 def __init__(self, methodName="runTest"):
     super(MobileBertSquadTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(
         MobileBertSquad, exported_names=["predict"])

Exemplo n.º 14

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(ConvTransposeModule)

Exemplo n.º 15

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(QuantizationModule)

Exemplo n.º 16

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         StatefulRingBufferModule, exported_names=["predict"])

Exemplo n.º 17

Arquivo: math_test.py Projeto: NeoTim/iree

 def __init__(self, *args, **kwargs):
     super(MathTest, self).__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(MathModule)

Exemplo n.º 18

 def __init__(self, *args, **kwargs):
     super(LogicalOpsTest, self).__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(LogicalOpsModule)

Exemplo n.º 19

Arquivo: mandelbrot_test.py Projeto: xinan-jiang/iree

 def __init__(self, *args, **kwargs):
   super().__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(MandelbrotModule)

Exemplo n.º 20

Arquivo: simple_stateful_test.py Projeto: yashsehgal/iree

 def __init__(self, *args, **kwargs):
     super(StatefulTest, self).__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(SimpleStatefulModule)

Exemplo n.º 21

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         LstmStaticModule, exported_names=["predict"])

Exemplo n.º 22

 def __init__(self, methodName="runTest"):
   super(MatrixOpsStaticTest, self).__init__(methodName)
   self._modules = tf_test_utils.compile_tf_module(MatrixOpsStaticModule)

Exemplo n.º 23

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(StringsModule)

Exemplo n.º 24

 def __init__(self, methodName="runTest"):
     super(SlimVisionTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(
         SlimVisionModule, exported_names=['predict'])

Exemplo n.º 25

Arquivo: einsum_dynamic_test.py Projeto: yashsehgal/iree

 def __init__(self, *args, **kwargs):
     super(EinsumDynamicTest, self).__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(EinsumDynamicModule)

Exemplo n.º 26

Arquivo: classification_training_test.py Projeto: shawnjohnjr/iree

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         ClassificationTrainingModule, exported_names=["train_on_batch"])

Exemplo n.º 27

Arquivo: sliding_window_test.py Projeto: xinan-jiang/iree

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         SlidingWindowModule, exported_names=["predict"])

Exemplo n.º 28

Arquivo: slim_vision_model_test.py Projeto: shawnjohnjr/iree

 def __init__(self, *args, **kwargs):
   super().__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(SlimVisionModule,
                                                   exported_names=['predict'])

Exemplo n.º 29

Arquivo: linspace_test.py Projeto: BernhardRiemann/iree

 def __init__(self, methodName="runTest"):
     super(LinspaceTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(LinspaceModule)

Exemplo n.º 30

 def __init__(self, methodName="runTest"):
     super(DynamicMlpTest, self).__init__(methodName)
     self._modules = tf_test_utils.compile_tf_module(
         DynamicMlpModule, exported_names=["predict"])