B ӻd@sdZddlZddlmZddlmZddlmZddlmZddd gZ dd d Z d d Z Gdddej Z Gdddej ZejjdZejejee efdZejZejZejZdS)zHThis module customizes `test_combinations` for `tf.keras` related tests.N)tf2) combinations)test_combinations) testing_utilsZ functionalsubclass sequentialcCsp|dkrtrdgnddg}|dkr.ddg}g}d|krN|tjdg|d7}d|krl|tjdgdgd7}|S)aReturns the default test combinations for tf.keras tests. Note that if tf2 is enabled, then v1 session test will be skipped. Args: mode: List of modes to run the tests. The valid options are 'graph' and 'eager'. Default to ['graph', 'eager'] if not specified. If a empty list is provide, then the test will run under the context based on tf's version, eg graph for v1 and eager for v2. run_eagerly: List of `run_eagerly` value to be run with the tests. Default to [True, False] if not specified. Note that for `graph` mode, run_eagerly value will only be False. Returns: A list contains all the combinations to be used to generate test cases. NeagergraphTF)mode run_eagerly)renabledrcombine)r r resultrV/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/keras/combinations.pykeras_mode_combinationssrcCs tjtdS)N) model_type)rr KERAS_MODEL_TYPESrrrrkeras_model_type_combinations8src@s eZdZdZddZddZdS)KerasModeCombinationzfCombination for Keras test mode. It by default includes v1_session, v2_eager and v2_tf_function. cCs(|dd}|dk r t|gSgSdS)Nr )poprZrun_eagerly_scope)selfkwargsr rrrcontext_managersBs  z%KerasModeCombination.context_managerscCs tdgS)Nr )rOptionalParameter)rrrrparameter_modifiersJsz(KerasModeCombination.parameter_modifiersN)__name__ __module__ __qualname____doc__rrrrrrr<src@s eZdZdZddZddZdS)KerasModelTypeCombinationaKCombination for Keras model types when doing model test. It by default includes 'functional', 'subclass', 'sequential'. Various methods in `testing_utils` to get models will auto-generate a model of the currently active Keras model type. This allows unittests to confirm the equivalence between different Keras models. cCs(|dd}|tkr t|gSgSdS)Nr)rrrZmodel_type_scope)rrrrrrrXs  z*KerasModelTypeCombination.context_managerscCs tdgS)Nr)rr)rrrrr_sz-KerasModelTypeCombination.parameter_modifiersN)rrrrrrrrrrr Nsr r)NN)r functoolsZtensorflow.pythonrZtensorflow.python.frameworkrrZtensorflow.python.kerasrrrrZTestCombinationrr generatekeywords _defaultspartialr timesZ NamedObjectrrrrs$