Home
last modified time | relevance | path

Searched refs:true_pred (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp1638 bool true_pred(Node* n) { in true_pred() function
1661 LoopsPeeler peeler(true_pred, 1); in TEST()
1675 LoopsPeeler peeler(true_pred, 3); in TEST()
1703 LoopsPeeler peeler(true_pred, 1); in TEST()
1717 LoopsPeeler peeler(true_pred, 3); in TEST()
1750 LoopsPeeler peeler(true_pred, 5); in TEST()
1764 LoopsPeeler peeler(true_pred, 1); in TEST()
1794 LoopsPeeler peeler(true_pred, 1); in TEST()
1805 LoopsPeeler peeler(true_pred, 5); in TEST()
1833 LoopsPeeler peeler(true_pred, 1); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconditional_test.cc507 auto true_pred = ConstantR0<bool>(&true_builder, true); in XLA_TEST_F() local
510 Tuple(&true_builder, {true_pred, true_scalar, true_array}); in XLA_TEST_F()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_higher_order_ops.py6480 true_pred = torch.Tensor([True])
6482 self.assertTrue(torch.allclose(test(true_pred, inp), opt_test(true_pred, inp)))