- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 56 for var_handle_op (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
StringRef variable_name = GetNodeNameFromClassAttrOrSharedNameAttr(variable_v2_op); if (variable_name.empty()) { return signalPassFailure(); } VarHandleOp var_handle_op = builder.create<VarHandleOp>( variable_v2_op.getLoc(), ArrayRef<Type>{RankedTensorType::get( {}, TF::ResourceType::get(ArrayRef<TensorType>{tensor_type},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
TEST(ShapeInferenceTest, WhileLoopWithResource) { // Graph: // x = resource_variable_ops.var_handle_op(dtype=dtypes.float32, shape=[2, 3]) // y = control_flow_ops.while_loop(lambda _: true, lambda x: x, [x]) Graph graph(OpRegistry::Global()); { Scope scope = Scope::NewRootScope().ExitOnError(); auto x = ops::VarHandleOp(scope.WithOpName("x"), DT_FLOAT, TensorShape({2, 3})); auto enter =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
namespace { Node* MakeRead(const Scope& scope, const string& id, Node** var_handle_op = nullptr) { Output var_handle = ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({})); Output read = ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT); if (var_handle_op) { *var_handle_op = var_handle.node(); } return read.node(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K 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/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)