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"])