- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 142 for var_handle_op (0.32 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_save_op.mlir
// CHECK: %[[VAR_HANDLE_0:.*]] = "tf.VarHandleOp"() // CHECK-SAME: {{.*shared_name = "var_0".*}} // CHECK: %[[VAR_HANDLE_1:.*]] = "tf.VarHandleOp"() // CHECK-SAME: {{.*shared_name = "var_1".*}} // ReadVariableOps are inserted for each VarHandleOp to read the tensor values. // CHECK-DAG: %[[READ_VARIABLE_0:.*]] = "tf.ReadVariableOp"(%[[VAR_HANDLE_0]]) : (tensor<!tf_type.resource<tensor<2xf32>>>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
if (auto pop_back = dyn_cast<TF::TensorListPopBackOp>(op)) { return InferShapeForTensorListPopBackOp(pop_back); } if (auto var_handle_op = dyn_cast<VarHandleOp>(op)) { return InferShapeForVarHandleOp(var_handle_op); } if (auto xla_reduce_window_op = dyn_cast<XlaReduceWindowOp>(op)) { return InferShapeForXlaReduceWindowOp(xla_reduce_window_op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
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/tfrt/tests/sink_in_invariant_ops.mlir
attributes {tf_saved_model.exported_names = ["main"]} { // CHECK: tf.VarHandleOp %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "", shared_name = "variable1"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>> %1 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "", shared_name = "variable2"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_mark_initialized_variables.mlir
%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", device = "/job:worker/replica:0/task:1/device:CPU:0", shared_name = "var1"} : () -> tensor<!tf_type.resource<tensor<100x50xf32>>> %1 = "tf.ReadVariableOp"(%0) {device = ""} : (tensor<!tf_type.resource<tensor<100x50xf32>>>) -> tensor<100x50xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/freeze_variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/promote_var_handles_to_args.mlir
// CHECK-NOT: "tf.VarHandleOp" func.func @no_args() { %0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<f32>>> func.return } // CHECK-LABEL: func @some_args // CHECK-SAME: (%arg0: tensor<i1>, %arg1: tensor<!tf_type.resource<tensor<f32>>> {tf.resource_name = "x"}) // CHECK-NOT: "tf.VarHandleOp" func.func @some_args(%arg0: tensor<i1>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/localize_var_handles.mlir
attributes {tf_saved_model.exported_names = ["handles_ambiguous_var_handles"]} { // CHECK: VarHandleOp %0 = "tf.VarHandleOp"() {container = "", shared_name = "foo"} : () -> tensor<!tf_type.resource<tensor<10xf32>>> func.call @assign_resource(%0) : (tensor<!tf_type.resource<tensor<10xf32>>>) -> () %1 = "tf.VarHandleOp"() {container = "", shared_name = "foo"} : () -> tensor<!tf_type.resource<tensor<10xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 23 21:12:02 UTC 2023 - 10.3K bytes - Viewed (0)