- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for while_node (0.16 sec)
-
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
EXPECT_EQ(input_node->name(), "arg0"); // Check node "while" input and output edges. const Node *while_node = node_name_index.at("while"); ASSERT_NE(while_node, nullptr); TF_CHECK_OK(while_node->input_node(0, &input_node)); EXPECT_EQ(input_node->name(), "arg1"); TF_CHECK_OK(while_node->input_node(1, &input_node)); EXPECT_EQ(input_node->name(), "arg0"); const Node *ret2_node = node_name_index.at("ret2");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
is_compilable &= ExtractNodeDefAndCheckCompilability( while_node, "cond", "while_cond", encapsulating_function, lib_runtime, stack_trace, uncompilable_nodes); if (!uncompilable_nodes && !is_compilable) return is_compilable; is_compilable &= ExtractNodeDefAndCheckCompilability( while_node, "body", "while_body", encapsulating_function, lib_runtime, stack_trace, uncompilable_nodes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
TF_RETURN_IF_ERROR(GraphToFunctionDef(*g, new_name, &replace_fdef)); TF_RETURN_IF_ERROR(fld->AddFunctionDef(replace_fdef)); loop_cond_func->set_name(new_name); while_node->ClearAttr("cond"); while_node->AddAttr("cond", *loop_cond_func); } return absl::OkStatus(); } // Rewrites while loop cond function for host. Status RewriteHostWhileLoopCond(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
NameAttrList* encapsulating_function, UncompilableNodesMap* uncompilable_nodes) const; bool IsCompilableWhile(const Node& while_node, FunctionLibraryRuntime* lib_runtime, std::vector<StackFrameView>* stack_trace, NameAttrList* encapsulating_function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_sequencing.mlir
// CHECK: {{.*tf.While.*body = @while_body.* cond = @while_cond.* parallel_iterations = 3}} %0 = "tf.While"(%cst) {body = @while_body, cond = @while_cond, is_stateless = false, parallel_iterations = 3} : (tensor<i32>) -> (tensor<i32>) return } func.func private @while_body(%arg0: tensor<i32>) -> (tensor<i32>) { // metadata ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 01 21:27:49 UTC 2023 - 19.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_pipelining.mlir
%0 = "tf.While"(%cst_main) {body = @while_body, cond = @while_cond, is_stateless = false} : (tensor<i32>) -> (tensor<i32>) // expected-error @+1 {{'tf.While' op multiple users of function.}} %1 = "tf.While"(%cst_main) {body = @while_body, cond = @while_cond, is_stateless = false} : (tensor<i32>) -> (tensor<i32>) return } func.func private @while_body(%arg0: tensor<i32>) -> (tensor<i32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 33.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/cluster_tf_ops_pass.mlir
func.return %1 : tensor<i1> } // The @while_body function is a Multi-hosts function which contains three // parts: // - A local subgraph which contains both local ops and remote_run kernels to // trigger remote subgraphs // - Two remote subgraph which contains remote ops on worker:1 and worker:2. // CHECK: func @while_body(%[[ARG_0:.*]]: tensor<i32> {tf.device = "/job:localhost/replica:0/task:0/device:CPU:0"})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/async_while.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 22.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir
// CHECK: func private @while_body // CHECK-LABEL: testWhileRegionTrivialCasts func.func private @while_cond(%arg0 : tensor<4xf32>, %arg1 : tensor<i32>) -> tensor<i1> func.func private @while_body(%arg0 : tensor<4xf32>, %arg1 : tensor<i32>) -> (tensor<4xf32>, tensor<i32>) func.func @testWhileRegionTrivialCasts(%arg0 : tensor<*xf32>, %arg1 : tensor<i32>) -> tensor<*xf32> { // CHECK: [[Result:%.*]]:2 = "tf.While"(%arg0, %arg1) <{body = @while_body, cond = @while_cond
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 02 11:15:34 UTC 2024 - 44.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/xla_launch.mlir
%0 = "tf.Const"() {value = dense<9> : tensor<i32>} : () -> tensor<i32> %1 = "tf.Less"(%arg0, %0) {} : (tensor<i32>, tensor<i32>) -> tensor<i1> func.return %1 : tensor<i1> } func.func @while_body(%arg0: tensor<i32>) -> tensor<i32> { %1 = "tf.AddV2"(%arg0, %arg0) {} : (tensor<i32>, tensor<i32>) -> tensor<i32> func.return %1 : tensor<i32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.6K bytes - Viewed (0)