Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1745 void CheckAddedLayers(OpConverterTest* test, bool expect_scale_layer) { in CheckAddedLayers() function
1783 CheckAddedLayers(test, /*expect_scale_layer=*/true); in TestBinaryTensorOpWeightNoBroadcast()
1838 CheckAddedLayers(test, /*expect_scale_layer=*/true); in TestBinaryTensorOpWeightWithChannelWiseBroadcast()
1873 CheckAddedLayers(test, /*expect_scale_layer=*/true); in TestBinaryTensorOpWeightWithUniformlyBroadcast()
1912 CheckAddedLayers(test, /*expect_scale_layer=*/false); in TestBinaryTensorOpWeightFallback()
1963 CheckAddedLayers(test, /*expect_scale_layer=*/false); in TestBinaryTensorOpTensor()