Searched refs:TestMultiPartition (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 285 void TestMultiPartition(bool gather_v2, bool include_shared_init, in TestMultiPartition() function in tensorflow::graph_transforms::SparsifyGatherTest
624 TEST_F(SparsifyGatherTest, TestMultiPartition) { in TEST_F() argument
625 TestMultiPartition(false, false, false); in TEST_F()
626 TestMultiPartition(false, true, false); in TEST_F()
627 TestMultiPartition(true, false, false); in TEST_F()
628 TestMultiPartition(true, true, false); in TEST_F()
629 TestMultiPartition(false, false, true); in TEST_F()
630 TestMultiPartition(false, true, true); in TEST_F()
631 TestMultiPartition(true, false, true); in TEST_F()
632 TestMultiPartition(true, true, true); in TEST_F()
[all …]
|