Searched refs:get_unpack_nodedef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6028 auto get_unpack_nodedef = [](DataType dtype, int num, int axis) -> NodeDef { in TEST_P() variable
6091 get_unpack_nodedef(dtype, ok_params[i].num, ok_params[i].axis); in TEST_P()
6130 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/3, /*axis=*/3); in TEST_P()
6139 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/1, /*axis=*/4); in TEST_P()
6148 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/1, /*axis=*/-5); in TEST_P()
6157 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/1, /*axis=*/0); in TEST_P()
6166 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/5, /*axis=*/2); in TEST_P()
6175 NodeDef node_def = get_unpack_nodedef(DT_FLOAT, /*num=*/1, /*axis=*/0); in TEST_P()
|