Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2510 auto get_squeeze_nodedef = [](std::vector<int> axis) -> NodeDef { in TEST_F() local
2523 NodeDef node_def = get_squeeze_nodedef({0}); in TEST_F()
2532 NodeDef node_def = get_squeeze_nodedef({0}); in TEST_F()
2540 NodeDef node_def = get_squeeze_nodedef({-4}); in TEST_F()
2548 NodeDef node_def = get_squeeze_nodedef({4}); in TEST_F()
2558 NodeDef node_def = get_squeeze_nodedef({-5}); in TEST_F()
2588 NodeDef node_def = get_squeeze_nodedef(ok_params[i].axis); in TEST_F()