Searched refs:ADD_OP (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2799 #define ADD_OP(name, op, v1, v2, v3, v4, v5, v6, v7, v8) \ in TEST_P() macro
2803 ADD_OP("Add", ops::Add, {5, 8, 6, 9, 5, 8, 6, 9}); in TEST_P()
2804 ADD_OP("AddV2", ops::AddV2, {5, 8, 6, 9, 5, 8, 6, 9}); in TEST_P()
2805 ADD_OP("Sub", ops::Sub, {1, 4, 0, 3, 1, 4, 0, 3}); in TEST_P()
2806 ADD_OP("Mul", ops::Mul, {6, 12, 9, 18, 6, 12, 9, 18}); in TEST_P()
2807 ADD_OP("Div", ops::Div, {1.5, 3, 1, 2, 1.5, 3, 1, 2}); in TEST_P()
2808 ADD_OP("RealDiv", ops::RealDiv, {1.5, 3, 1, 2, 1.5, 3, 1, 2}); in TEST_P()
2809 ADD_OP("FloorDiv", ops::FloorDiv, {1, 3, 1, 2, 1, 3, 1, 2}); in TEST_P()
2810 ADD_OP("Minimum", ops::Minimum, {2, 2, 3, 3, 2, 2, 3, 3}); in TEST_P()
2811 ADD_OP("Maximum", ops::Maximum, {3, 6, 3, 6, 3, 6, 3, 6}); in TEST_P()
[all …]
|