Searched refs:c_mat (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | where_op_test.py | 263 c_mat = np.array([[False] * 192, [True] * 192] * 8192) # [16384, 192]
265 np_val = np.where(c_mat, x * x, -x)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 536 Output c_mat = ops::Const(s.WithOpName("c_mat"), 2.0f, {2, 2}); in TEST_F() local
544 Output child1 = ops::BiasAdd(s.WithOpName("child1"), c_mat, x_vec); in TEST_F()
546 Output child1a = ops::BiasAdd(s.WithOpName("child1a"), c_mat, x_vec); in TEST_F()
550 Output parent2 = ops::Add(s.WithOpName("parent2"), child2, c_mat); in TEST_F()
552 Output parent2a = ops::Add(s.WithOpName("parent2a"), c_mat, child2a); in TEST_F()
554 Output child3 = ops::Add(s.WithOpName("child3"), c_mat, x_vec); in TEST_F()
556 Output child3a = ops::Add(s.WithOpName("child3a"), x_vec, c_mat); in TEST_F()
559 Output child4 = ops::BiasAdd(s.WithOpName("child4"), c_mat, x_vec); in TEST_F()
564 Output parent5 = ops::BiasAdd(s.WithOpName("parent5"), c_mat, child5); in TEST_F()
566 Output parent6 = ops::BiasAdd(s.WithOpName("parent6"), c_mat, child6); in TEST_F()
|