Exemplos de compile_tf_module em Python

Exemplo n.º 1

Arquivo: keyword_spotting_streaming_test.py Projeto: 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))

Exemplo n.º 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))

Exemplo n.º 3

Arquivo: fft_test.py Projeto: 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)

Exemplo n.º 4

Arquivo: einsum_vector_test.py Projeto: julianwa/iree

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

Exemplo n.º 5

Arquivo: sliding_window_test.py Projeto: sifive/iree

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

Exemplo n.º 6

Arquivo: quantization_dyn_test.py Projeto: sifive/iree

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

Exemplo n.º 7

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

Exemplo n.º 8

Arquivo: layers_test.py Projeto: 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())

Exemplo n.º 9

Arquivo: image_resize_test.py Projeto: julianwa/iree

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

Exemplo n.º 10

Arquivo: tensorlist_test.py Projeto: julianwa/iree

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

Exemplo n.º 11

Arquivo: ring_buffer_test.py Projeto: julianwa/iree

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

Exemplo n.º 12

Arquivo: applications_test.py Projeto: 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))

Exemplo n.º 13

Arquivo: depth_conv_test.py Projeto: sifive/iree

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

Exemplo n.º 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)

Exemplo n.º 15

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

Exemplo n.º 16

Arquivo: tf_test_utils_test.py Projeto: sifive/iree

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

Exemplo n.º 17

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

Exemplo n.º 18

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

Exemplo n.º 19

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

Exemplo n.º 20

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

Exemplo n.º 21

Arquivo: dynamic_mlp_relu_test.py Projeto: julianwa/iree

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

Exemplo n.º 22

Arquivo: space_to_batch_nd_test.py Projeto: julianwa/iree

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

Exemplo n.º 23

Arquivo: mandelbrot_test.py Projeto: sifive/iree

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

Exemplo n.º 24

Arquivo: broadcast_to_test.py Projeto: julianwa/iree

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

Exemplo n.º 25

Arquivo: simple_stateful_test.py Projeto: sifive/iree

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