Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc5855 auto get_split_nodedef = [](DataType dtype, int num_split) -> NodeDef { in TEST_P() variable
5912 NodeDef node_def = get_split_nodedef(dtype, ok_params[i].num_split); in TEST_P()
5952 NodeDef node_def = get_split_nodedef(DT_FLOAT, 1); in TEST_P()
5962 NodeDef node_def = get_split_nodedef(DT_FLOAT, 1); in TEST_P()
5972 NodeDef node_def = get_split_nodedef(DT_FLOAT, 1); in TEST_P()
5982 NodeDef node_def = get_split_nodedef(DT_FLOAT, 1); in TEST_P()
5992 NodeDef node_def = get_split_nodedef(DT_FLOAT, 1); in TEST_P()
6002 NodeDef node_def = get_split_nodedef(DT_FLOAT, 2); in TEST_P()
6012 NodeDef node_def = get_split_nodedef(DT_FLOAT, 4); in TEST_P()