- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 142 for var_handle_op (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
to input arguments and outputs of the function(s). Two types of resources are supported: (1) A function argument of TF::ResourceType type (this pass). (2) A VarHandleOp in the function (tf-promote-var-handles-to-args). After the pass, . The function will have an input argument for each resource that is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
%island = tf_executor.island { // expected-remark@above {{ID: 10}} // expected-remark@above {{Successors: {11}}} %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> // expected-remark@above {{ID: 0}} %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> // expected-remark@above {{ID: 1}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
if the variable is read only. For example, the following pseudo MLIR code (types are left out for brevity) ```mlir func.func @hoist_loop_invariant(%arg0, %arg1) { %var = "tf.VarHandleOp"() {container="", shared_name="var_name", device = "/device:CPU:0"} %results:2 = "tf.WhileRegion"(%arg0, %arg1) ({ ^bb0(%arg2, %arg3): %0 = "tf.OpA"() {is_stateless = true} "tf.Yield"(%0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// aliasing output arguments. std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass( llvm::ArrayRef<std::string> functions = {}); // Creates a pass that promotes tf.VarHandleOp to resource arguments for all // functions. std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass(); // Creates a pass that converts readonly reference variables to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-dynamic-layout-pass.mlir
tf_device.return %1#0, %1#1 : tensor<!tf_type.string>, tensor<2x!tf_type.string> }) {device = "/device:CPU:0"} : () -> (tensor<!tf_type.string>, tensor<2x!tf_type.string>) %var = "tf.VarHandleOp"() {container = "c", shared_name = "v", device = "/device:TPU:0"} : () -> tensor<!tf_type.resource<tensor<3x3x1x32xf32>>> // CHECK-NOT: "tf.TPUGetLayoutOp" // CHECK-NOT: "tf.TPUCopyWithLayout"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
return VerifyUnsortedSegmentReduction(*this); } //===----------------------------------------------------------------------===// // VarHandleOp //===----------------------------------------------------------------------===// LogicalResult VarHandleOp::verify() { // VarHandleOp requires the resource handle supply a single subtype from // which to derive the dtype and shape attributes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir
%outputs_8 = "tf.Const"() {device = "/job:localhost/replica:0/task:0/device:CPU:0", value = dense<1> : tensor<i32>} : () -> tensor<i32> // CHECK: [[elems:%.*]] = "tf.VarHandleOp" %outputs_10 = "tf.VarHandleOp"() {_xla_inferred_shapes = [#tf_type.shape<>], allowed_devices = [], container = "", device = "/job:localhost/replica:0/task:0/device:CPU:0", shared_name = "w"} : () -> tensor<!tf_type.resource<tensor<3x1xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.return } // ----- // Valid VarHandleOp operation. // CHECK-LABEL: func @testVarHandleOp func.func @testVarHandleOp() -> tensor<!tf_type.resource<tensor<*xf32>>> { %0 = "tf.VarHandleOp"() { container = "", shared_name = "cd2c89b7-88b7-44c8-ad83-06c2a9158347" } : () -> tensor<!tf_type.resource<tensor<*xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
func.return %0 : tensor<*xf32> } // CHECK-LABEL: @erase_tf_var_is_initialized func.func @erase_tf_var_is_initialized(%arg0 : tensor<!tf_type.resource<tensor<f32>>>) -> tensor<i1> { %vh = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<f32>>> %is = "tf.VarIsInitializedOp"(%vh) : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
if (std::optional<absl::string_view> cluster_name = GetXlaClusterForNode(*n)) { n->set_name(absl::StrCat(*cluster_name, "/", n->name())); } else if (n->type_string() == "VarHandleOp") { n->set_name(absl::StrCat("varhandle/", n->name())); } else { // There is room for improvement here. In particular, it may help to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)