- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for var_handle_op (0.22 sec)
-
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/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/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/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/ir/tf_generated_ops.td
See `tf.scatter_nd` for more details about how to make updates to slices. }]; let arguments = (ins Arg<TF_ResourceTensor, [{A resource handle. Must be from a VarHandleOp.}], [TF_VariableRead, TF_VariableWrite]>:$ref, Arg<TF_I32OrI64Tensor, [{A Tensor. Must be one of the following types: int32, int64. A tensor of indices into ref.}]>:$indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)