- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for while_loop_test (0.13 sec)
-
tensorflow/cc/ops/while_loop_test.cc
#include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { class WhileLoopTest : public ::testing::Test { protected: WhileLoopTest() : scope_(Scope::NewRootScope()) {} void Init(int num_inputs, DataType dtype = DT_INT32) { for (int i = 0; i < num_inputs; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/device_copy.mlir
func.return %outputs_0, %outputs_1 : tensor<2x2xf32>, tensor<2x2xf32> } // CHECK: func @while_loop_test(%[[ARG_0:.*]]: tensor<i32>, %[[ARG_1:.*]]: tensor<i32>, %arg2: tensor<*xf32>) func.func @while_loop_test(%arg0: tensor<i32>, %arg1: tensor<i32>, %arg2: tensor<*xf32>) { // CHECK-NEXT: tf.WhileRegion %0:2 = "tf.WhileRegion"(%arg0, %arg2) ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 5.2K bytes - Viewed (0)