Searched refs:assert_num_traces (Results 1 – 1 of 1) sorted by relevance
955 def assert_num_traces(layer_cls, training_keyword): function977 assert_num_traces(LayerWithTrainingKeyword, training_keyword=True)984 assert_num_traces(LayerWithKwargs, training_keyword=False)995 assert_num_traces(LayerWithChildLayer, training_keyword=False)