Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc6372 NodeDef GetArgMinMaxNodeDef(DataType input_dtype, DataType output_dtype) { in TEST_P() function
6396 NodeDef node_def = GetArgMinMaxNodeDef<OpType>(_tf_type, in TEST_P()
6422 GetArgMinMaxNodeDef<ops::ArgMax>(tf_type_, in TEST_P()
6434 GetArgMinMaxNodeDef<ops::ArgMax>(tf_type_, in TEST_P()