Python compile_tf_moduleの例

コード例 #1

ファイル: keyword_spotting_streaming_test.py プロジェクト: 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))

コード例 #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))

コード例 #3

ファイル: fft_test.py プロジェクト: 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)

コード例 #4

ファイル: einsum_vector_test.py プロジェクト: julianwa/iree

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

コード例 #5

ファイル: sliding_window_test.py プロジェクト: sifive/iree

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

コード例 #6

ファイル: quantization_dyn_test.py プロジェクト: sifive/iree

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

コード例 #7

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

コード例 #8

ファイル: layers_test.py プロジェクト: 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())

コード例 #9

ファイル: image_resize_test.py プロジェクト: julianwa/iree

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

コード例 #10

ファイル: tensorlist_test.py プロジェクト: julianwa/iree

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

コード例 #11

ファイル: ring_buffer_test.py プロジェクト: julianwa/iree

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

コード例 #12

ファイル: applications_test.py プロジェクト: 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))

コード例 #13

ファイル: depth_conv_test.py プロジェクト: sifive/iree

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

コード例 #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)

コード例 #15

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

コード例 #16

ファイル: tf_test_utils_test.py プロジェクト: sifive/iree

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

コード例 #17

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

コード例 #18

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

コード例 #19

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

コード例 #20

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

コード例 #21

ファイル: dynamic_mlp_relu_test.py プロジェクト: julianwa/iree

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

コード例 #22

ファイル: space_to_batch_nd_test.py プロジェクト: julianwa/iree

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

コード例 #23

ファイル: mandelbrot_test.py プロジェクト: sifive/iree

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

コード例 #24

ファイル: broadcast_to_test.py プロジェクト: julianwa/iree

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

コード例 #25

ファイル: simple_stateful_test.py プロジェクト: sifive/iree

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