Python compile_tf_module Examples

Example #1

File: keyword_spotting_streaming_test.py Project: sifive/iree

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         KeywordSpottingModule,
         exported_names=['call'],
         relative_artifacts_dir=os.path.join('kws_streaming', FLAGS.model,
                                             FLAGS.mode))

Example #2

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

Example #3

File: fft_test.py Project: sifive/iree

    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)

Example #4

File: einsum_vector_test.py Project: julianwa/iree

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

Example #5

File: sliding_window_test.py Project: sifive/iree

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

Example #6

File: quantization_dyn_test.py Project: sifive/iree

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

Example #7

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

Example #8

File: layers_test.py Project: julianwa/iree

 def __init__(self, *args, **kwargs):
     super().__init__(*args, **kwargs)
     self._modules = tf_test_utils.compile_tf_module(
         KerasLayersModule,
         exported_names=KerasLayersModule.get_tf_function_unit_tests(),
         relative_artifacts_dir=get_relative_artifacts_dir())

Example #9

File: image_resize_test.py Project: julianwa/iree

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

Example #10

File: tensorlist_test.py Project: julianwa/iree

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

Example #11

File: ring_buffer_test.py Project: julianwa/iree

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

Example #12

File: applications_test.py Project: sifive/iree

 def __init__(self, *args, **kwargs):
   super().__init__(*args, **kwargs)
   self._modules = tf_test_utils.compile_tf_module(
       ApplicationsModule,
       exported_names=ApplicationsModule.get_tf_function_unit_tests(),
       relative_artifacts_dir=os.path.join(FLAGS.model, FLAGS.data))

Example #13

File: depth_conv_test.py Project: sifive/iree

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

Example #14

 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)

Example #15

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

Example #16

File: tf_test_utils_test.py Project: sifive/iree

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

Example #17

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

Example #18

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

Example #19

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

Example #20

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

Example #21

File: dynamic_mlp_relu_test.py Project: julianwa/iree

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

Example #22

File: space_to_batch_nd_test.py Project: julianwa/iree

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

Example #23

File: mandelbrot_test.py Project: sifive/iree

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

Example #24

File: broadcast_to_test.py Project: julianwa/iree

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

Example #25

File: simple_stateful_test.py Project: sifive/iree

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