Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for var_handle_op (0.18 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top