Searched refs:ExpectNodeEqual (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 27 void ExpectNodeEqual(const Node* n, gtl::ArraySlice<T> values, in ExpectNodeEqual() function
56 ExpectNodeEqual<float>(c.node(), {42.0f}, {}); in TEST()
64 ExpectNodeEqual<double>(c.node(), {2.0, 3.0}, {2, 1}); in TEST()
95 ExpectNodeEqual<double>(c.node(), {42.0, 42.0, 42.0, 42.0}, {2, 2}); in TEST()
100 ExpectNodeEqual<string>(d.node(), {"1", "2", "3", "4", "5", "6"}, {2, 3}); in TEST()
114 ExpectNodeEqual<double>(c.node(), {0.0, 1.0, 2.0, 3.0}, {2, 2}); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 64 void ExpectNodeEqual(const Node* n, gtl::ArraySlice<T> values, in ExpectNodeEqual() function in tensorflow::__anond28d22700111::ConstantFoldingTest
265 ExpectNodeEqual<int>(*(b0->in_nodes().begin()), {0, 1}, {2}); in TEST_F()
267 ExpectNodeEqual<int>(*(b1->in_nodes().begin()), {}, {0}); in TEST_F()
299 ExpectNodeEqual<int>(*(b0->in_nodes().begin()), {0, 1}, {2}); in TEST_F()
306 ExpectNodeEqual<int>(*(b1_ident->in_nodes().begin()), {}, {0}); in TEST_F()
434 ExpectNodeEqual<int>(p, {3}, {}); in TEST_F()
496 ExpectNodeEqual<int>(cf0, {26}, {}); in TEST_F()
500 ExpectNodeEqual<int>(cf1, {2, 4}, {2}); in TEST_F()
504 ExpectNodeEqual<int>(cf2, {4, 6, 8}, {3}); in TEST_F()
576 ExpectNodeEqual<int>(cf0, {2}, {}); in TEST_F()
[all …]
|