Searched refs:GetDepthSpaceShuffleNodeDef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6539 NodeDef GetDepthSpaceShuffleNodeDef(DataType dtype, int block_size, in TEST_P() function
6565 NodeDef node_def = GetDepthSpaceShuffleNodeDef<OpType>( in TEST_P()
6644 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6654 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6664 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 3, "NCHW"); in TEST_P()
6673 NodeDef node_def = GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>( in TEST_P()
6743 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6753 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6763 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 3, "NCHW"); in TEST_P()
6773 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 3, "NCHW"); in TEST_P()
[all …]
|