Searched refs:v_ctrl_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner_test.cc | 329 Tensor v_ctrl_t(DT_BOOL, TensorShape({})); in TEST_F() local
330 v_ctrl_t.flat<bool>()(0) = true; in TEST_F()
332 item.graph, {"merge", "id2"}, {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
334 {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
|
| D | constant_folding_test.cc | 1529 Tensor v_ctrl_t(DT_BOOL, TensorShape({})); in TEST_F() local
1531 v_ctrl_t.flat<bool>()(0) = true; in TEST_F()
1534 item.graph, fetch_nodes, {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
1537 {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
1542 v_ctrl_t.flat<bool>()(0) = false; in TEST_F()
1544 {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
1547 {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
1617 Tensor v_ctrl_t(DT_BOOL, TensorShape({})); in TEST_F() local
1618 v_ctrl_t.flat<bool>()(0) = true; in TEST_F()
1620 item.graph, item.fetch, {{"v_in", v_in_t}, {"v_ctrl", v_ctrl_t}}); in TEST_F()
[all …]
|