Searched refs:CreateReduceOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5461 NodeDef CreateReduceOp(DataType tf_type, bool keep_dims) { in CreateReduceOp() function
5490 const NodeDef node_def = CreateReduceOp<ops::Sum>(tf_type_, false); in TEST_P()
5500 const NodeDef node_def = CreateReduceOp<ops::Sum>(tf_type_, false); in TEST_P()
5516 {"Sum", CreateReduceOp<ops::Sum>, [](float x, float y) { return x + y; }, in TEST_P()
5518 {"Prod", CreateReduceOp<ops::Prod>, in TEST_P()
5520 {"Mean", CreateReduceOp<ops::Mean>, in TEST_P()
5522 {"Min", CreateReduceOp<ops::Min>, in TEST_P()
5524 {"Max", CreateReduceOp<ops::Max>, in TEST_P()
|