- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 41 for var_handle_op (1.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc
return signalPassFailure(); } auto var_handle_ops = init_func_op.getBlocks().front().getOps<VarHandleOp>(); llvm::SmallVector<VarHandleOp, 4> init_vars(var_handle_ops.begin(), var_handle_ops.end()); RemoveVariables(init_vars); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
shared_name: the name by which this variable is referred to. dtype and shape: attributes representing the data type and shape held in the variable. Example: resource_variable_ops.var_handle_op( dtype=dtypes.int32, shape=[8, 16], container="foo", shared_name="bar") returns a handle for a variable with name "bar" in container "foo", and the variable holds a tensor of shape [8, 16] and dtype int32. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc
ImmediateTensorHandlePtr* handle) { ImmediateOpPtr varhandle_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(varhandle_op->Reset("VarHandleOp", raw_device_name)); TF_RETURN_IF_ERROR(varhandle_op->SetAttrType("dtype", dtype)); // Note that if shape is unknown rank, shape.dim_sizes() will be empty, and // shape.dims() will be -1.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 11:28:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
// CHECK-DAG: [[VAR_HANDLE:%.*]] = "tf.VarHandleOp" // CHECK-DAG: [[MG_HANDLE:%.*]] = "tf.VarHandleOp" // CHECK-DAG: [[MS_HANDLE:%.*]] = "tf.VarHandleOp" // CHECK-DAG: [[MOM_HANDLE:%.*]] = "tf.VarHandleOp" %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<f32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
// CHECK-NOT: IfrtLoadVariable // CHECK: "tf.VarHandleOp" // CHECK-NEXT: "tf.AssignVariableOp" // CHECK-NEXT: "tf.ReadVariableOp" // CHECK-NEXT: "tf.StatefulPartitionedCall" // CHECK-NEXT: return // module { func.func @serving_default() -> tensor<*xi32> { %cst = "tf.Const"() <{value = dense<"some_test.txt"> : tensor<!tf_type.string>}> : () -> tensor<!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/lower_to_ifrt_restore_variable.mlir
%1 = "tf.VarHandleOp"() <{container = "", shared_name = "y"}> : () -> tensor<!tf_type.resource<tensor<3x1xf32>>> "tf.AssignVariableOp"(%1, %0) : (tensor<!tf_type.resource<tensor<3x1xf32>>>, tensor<3x1xf32>) -> () return } } // ----- // single variable: VarHandleOp is before RestoreV2 // CHECK-LABEL: func.func @varhandle_before_restore() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 8.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/rewrite_ifrt_load_variable.mlir
// Variable is used by both CPU and TPU // // CHECK-LABEL: func @serving_default(%arg0: tensor<1x3xf32>) -> tensor<1x1xf32> // CHECK-NEXT: [[HANDLE:%.*]] = "tf.VarHandleOp"() // CHECK-NEXT: [[ARRAYKEY:%.*]], [[FURTURE:%.*]] = "tf_mlrt.tf_ifrt_load_variable"([[HANDLE]]) // CHECK-SAME: <{used_by_host = true}> : (tensor<!tf_type.resource<tensor<3x1xf32>>>) -> (tensor<!tf_type.string>, !mlrt.future)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:35:32 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
// CHECK-NOT: "tf_saved_model.session_initializer" "tf_saved_model.session_initializer"() { initializers = [@func_init] } : () -> () // CHECK-LABEL: _tfrt_resource_init // CHECK: tf.VarHandleOp // CHECK: tf.ReadVariableOp // CHECK: tfrt_fallback_async.set_resource // CHECK-SAME: {device = "/device:CPU:0", index = 0 : i64} // CHECK-LABEL: func @init // CHECK-SAME: {tfrt.cost_threshold = 1 : i64}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/passes.h
// GraphDef. std::unique_ptr<OperationPass<ModuleOp>> CreateMergeSaveFunctionOpsToMainPass(); // Creates a pass that "unfreezes" ConstOps into variables. Each ConstOp's use // will be replaced by a VarHandleOp -> ReadVariableOp pattern. The newly // created variables will be initialized in the session initializer function via // AssignVariableOps. std::unique_ptr<OperationPass<ModuleOp>> CreateUnfreezeConstantsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 12.3K bytes - Viewed (0)