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)