- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for while_node (0.17 sec)
-
tensorflow/compiler/mlir/lite/tests/guarantee_func_has_one_use.mlir
%0 = "tf.While"(%arg0) {body = @while_body, cond = @while_cond, device = "", is_stateless = true} : (tensor<256x256xi32>) -> (tensor<256x256xi32>) %1 = "tf.While"(%arg1) {body = @while_body, cond = @while_cond, device = "", is_stateless = true} : (tensor<128xi32>) -> (tensor<128xi32>) func.return %0, %1: tensor<256x256xi32>, tensor<128xi32> } func.func @while_body(%arg0: tensor<*xi32>) -> (tensor<*xi32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 1.5K 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/tensorflow/tests/drop_while_shape_invariant.mlir
func.func @while_cond(%arg0: tensor<*xf32>) -> tensor<i1> { %0 = "tf.Const"() {value = dense<true> : tensor<i1>} : () -> tensor<i1> func.return %0 : tensor<i1> } func.func @while_body(%arg0: tensor<*xf32>) -> (tensor<*xf32>) { %0 = "tf.SomeOp"(%arg0) : (tensor<*xf32>) -> tensor<*xf32> func.return %0 : tensor<*xf32> } // Test that -tf-drop-while-shape-invariant-in-device-cluster pass does not drop
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2.6K 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) -
tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir
// CHECK: yield // CHECK: ^bb0([[ARGS]]): // CHECK: call @while_body // CHECK: yield // CHECK: while_body // CHECK: while_cond // INLINE: tfl.while // INLINE: ^bb0([[ARGS:.*]]): // INLINE: tf.Less // INLINE: yield // INLINE: ^bb0([[ARGS]]): // INLINE: %cst_2 = arith.constant // INLINE: yield // INLINE-NOT: while_body // INLINE-NOT: while_cond // CANON-LABEL: func @while_main
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.cc
// generated module. auto& child_node = nodes.at(child.node_id()); auto child_name = child.local_name().c_str(); if (child_node.kind_case() == tensorflow::SavedObject::kFunction) { Function tf_function; for (const std::string& fn_name : child_node.function().concrete_functions()) { // Setup input signature. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/end-to-end-tpu-reshard-variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlprogram.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 19:27:16 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/g3doc/space_to_depth.md
input. #### Pseudo MLIR code before and after RewritePass ```mlir // Example: original program: // module { func @while_body { %input = "tf.IteratorGetNext"(...) {device = "/CPU:0"}: -> tensor<2x224x224x3xf32> %device_launch = "tf_device.launch_func"(%input,...) {func = @_func,...) return ... }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Oct 24 02:51:43 UTC 2020 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-input-func-arg-name-collision.pbtxt
# This check that we don't error out when importing GraphDef containing # functions with arg name that are the same as the graph input name # CHECK: func @main(%arg0: tensor<{{.*}}i32>) -> tensor<{{.*}}i32> # CHECK: func private @while_body # CHECK: func private @while_cond node { name: "input" op: "Placeholder" attr { key: "_user_specified_name" value { s: "input" } } attr { key: "dtype"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 11 19:14:04 UTC 2020 - 4.8K bytes - Viewed (0)