Home
last modified time | relevance | path

Searched refs:assert_num_traces (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py955 def assert_num_traces(layer_cls, training_keyword): function
977 assert_num_traces(LayerWithTrainingKeyword, training_keyword=True)
984 assert_num_traces(LayerWithKwargs, training_keyword=False)
995 assert_num_traces(LayerWithChildLayer, training_keyword=False)