Searched refs:if_t (Results 1 – 1 of 1) sorted by relevance
4240 std::unique_ptr<Tensor> if_t; in TEST_F() local4242 if_t.reset(new Tensor(DT_BOOL, TensorShape())); in TEST_F()4244 if_t.reset(new Tensor(DT_BOOL, TensorShape({2, 2}))); in TEST_F()4247 if_t->flat<bool>()(i) = pred_val; in TEST_F()4249 Output if_ = ops::Const(scope.WithOpName("if"), *if_t); in TEST_F()4306 std::unique_ptr<Tensor> if_t; in TEST_F() local4307 if_t.reset(new Tensor(DT_BOOL, pred_shape)); in TEST_F()4309 if_t->flat<bool>()(i) = pred_val; in TEST_F()4311 Output if_ = ops::Const(scope.WithOpName("if"), *if_t); in TEST_F()