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))
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)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ResourcesOpsModule)
def __init__(self, *args, **kwargs):
super(DynamicMlpTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(DynamicMlpModule,
exported_names=["predict"])
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)
def __init__(self, methodName="runTest"):
super(LogicalOpsTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(LogicalOpsModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(DepthConv2dModule)
def __init__(self, *args, **kwargs): super(BatchNormTest, self).__init__(*args, **kwargs) self._modules = tf_test_utils.compile_tf_module(BatchNormModule)
def __init__(self, *args, **kwargs): super(ControlFlowTest, self).__init__(*args, **kwargs) self._modules = tf_test_utils.compile_tf_module(ControlFlowModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(BroadcastingModule)
def __init__(self, *args, **kwargs): super(SimpleArithmeticTest, self).__init__(*args, **kwargs) self._modules = tf_test_utils.compile_tf_module(SimpleArithmeticModule)
def __init__(self, methodName="runTest"):
super(BatchNormTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(BatchNormModule)
def __init__(self, methodName="runTest"):
super(MobileBertSquadTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(
MobileBertSquad, exported_names=["predict"])
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ConvTransposeModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(QuantizationModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(
StatefulRingBufferModule, exported_names=["predict"])
def __init__(self, *args, **kwargs):
super(MathTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MathModule)
def __init__(self, *args, **kwargs):
super(LogicalOpsTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(LogicalOpsModule)
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self._modules = tf_test_utils.compile_tf_module(MandelbrotModule)
def __init__(self, *args, **kwargs):
super(StatefulTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SimpleStatefulModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(
LstmStaticModule, exported_names=["predict"])
def __init__(self, methodName="runTest"): super(MatrixOpsStaticTest, self).__init__(methodName) self._modules = tf_test_utils.compile_tf_module(MatrixOpsStaticModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(StringsModule)
def __init__(self, methodName="runTest"):
super(SlimVisionTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(
SlimVisionModule, exported_names=['predict'])
def __init__(self, *args, **kwargs):
super(EinsumDynamicTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(EinsumDynamicModule)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(
ClassificationTrainingModule, exported_names=["train_on_batch"])
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(
SlidingWindowModule, exported_names=["predict"])
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SlimVisionModule,
exported_names=['predict'])
def __init__(self, methodName="runTest"):
super(LinspaceTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(LinspaceModule)
def __init__(self, methodName="runTest"):
super(DynamicMlpTest, self).__init__(methodName)
self._modules = tf_test_utils.compile_tf_module(
DynamicMlpModule, exported_names=["predict"])
