Searched refs:while_loop_test (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| H A D | device_copy.mlir | 68 // CHECK: func @while_loop_test(%[[ARG_0:.*]]: tensor<i32>, %[[ARG_1:.*]]: tensor<i32>, %arg2: tens…
69 func.func @while_loop_test(%arg0: tensor<i32>, %arg1: tensor<i32>, %arg2: tensor<*xf32>) {
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 1799 def test_while_loop_functionalize(self, func_type, while_loop_test): argument
1802 if while_loop_test not in ("simple_with_linear", "nested_with_linear"):
1803 fn, inp = WHILE_LOOP_TESTS[while_loop_test]
1810 def test_while_loop_tracing(self, while_loop_test): argument
1811 fn, inp = WHILE_LOOP_TESTS[while_loop_test]
1814 if while_loop_test not in ("simple_with_linear", "nested_with_linear")
1821 def test_while_loop_compile(self, backend, while_loop_test): argument
1822 fn, inp = WHILE_LOOP_TESTS[while_loop_test]
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | BUILD | 903 name = "while_loop_test",
905 srcs = ["while_loop_test.cc"],
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/ |
| H A D | BUILD | 329 srcs = ["ops/while_loop_test.cc"],
|