- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 41 for var_handle_op (0.75 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_cluster_formation.mlir
func.func @resource_before_cluster() { // CHECK-NEXT: [[CONST:%.*]] = "tf.Const" %0 = "tf.Const"() {value = dense<1.000000e+00> : tensor<f32>} : () -> tensor<f32> // CHECK-NEXT: [[UNUSED_RESOURCE:%.*]] = "tf.VarHandleOp" // CHECK-NEXT: "tf.AssignAddVariableOp"([[UNUSED_RESOURCE]], [[CONST]]) // CHECK: "tf_device.cluster" // CHECK-NEXT: "tf.NoOp" // CHECK-NEXT: tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 53.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
if (auto global = tf_saved_model::LookupBoundInputOfType< tf_saved_model::GlobalTensorOp>(func, i, syms)) { OpBuilder builder(func.getBody()); auto dummy = builder.create<TF::VarHandleOp>( global.getLoc(), func.getArgument(i).getType(), "dummy", "dummy"); func.getArgument(i).replaceAllUsesWith(dummy.getResult()); argsToErase.set(i); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir
func.return %arg : tensor<!tf_type.resource<tensor<i32>>> } // CHECK-LABEL: func @side_effect_while_test func.func @side_effect_while_test() -> (tensor<i32>) { %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "c", shared_name = "v"} : () -> tensor<!tf_type.resource<tensor<i32>>> // CHECK: [[while_res:%.]]:2 = tfrt.while {{%.*}} @"side_effect_while_body_add2/tfrt_body_1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K 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_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/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/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
Args: node_def: `NodeDef` to test whether it is a variable or not. Returns: Returns True if it is a variable. """ return node_def.op == 'VarHandleOp' def _find_variables( graph_def: graph_pb2.GraphDef, ) -> Mapping[str, node_def_pb2.NodeDef]: """Finds all variables within `graph_def`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0)