Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for var_handle (0.18 sec)

  1. tensorflow/compiler/mlir/lite/tests/quantize-variables.mlir

      %0 = "tfl.var_handle"() : () -> tensor<!tf_type.resource>
      %1 = "tfl.dequantize"(%arg0) : (tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>) -> tensor<1x2x1x3xf32>
      "tfl.assign_variable"(%0, %1) : (tensor<!tf_type.resource>, tensor<1x2x1x3xf32>) -> ()
      func.return %arg0 : tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 20.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/quantize.mlir

    // CHECK-NEXT: return %[[q]]
    }
    
    // CHECK-LABEL: NotQuantizeReadVariable
    func.func @NotQuantizeReadVariable() -> tensor<1x2x3x!quant.uniform<u8<1:255>:f32, 0.047244094488188976:128>> {
      %0 = "tfl.var_handle"() {container = "", shared_name = "states"} : () -> tensor<!tf_type.resource<tensor<1x2x3xf32>>>
      %1 = "tfl.read_variable"(%0) : (tensor<!tf_type.resource<tensor<1x2x3xf32>>>) -> tensor<1x2x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:10:13 UTC 2024
    - 39.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/schema/schema_v3b.fbs

      IMAG=133,
      REAL=134,
      COMPLEX_ABS=135,
      HASHTABLE = 136,
      HASHTABLE_FIND = 137,
      HASHTABLE_IMPORT = 138,
      HASHTABLE_SIZE = 139,
      REDUCE_ALL = 140,
      CONV_3D_TRANSPOSE = 141,
      VAR_HANDLE = 142,
      READ_VARIABLE = 143,
      ASSIGN_VARIABLE = 144,
    }
    // LINT.ThenChange(nnapi_linter/linter.proto)
    
    // Options for the builtin operators.
    union BuiltinOptions {
      Conv2DOptions,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 14:28:27 UTC 2024
    - 30K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/schema/schema.fbs

      IMAG=133,
      REAL=134,
      COMPLEX_ABS=135,
      HASHTABLE = 136,
      HASHTABLE_FIND = 137,
      HASHTABLE_IMPORT = 138,
      HASHTABLE_SIZE = 139,
      REDUCE_ALL = 140,
      CONV_3D_TRANSPOSE = 141,
      VAR_HANDLE = 142,
      READ_VARIABLE = 143,
      ASSIGN_VARIABLE = 144,
      BROADCAST_ARGS = 145,
      RANDOM_STANDARD_NORMAL = 146,
      BUCKETIZE = 147,
      RANDOM_UNIFORM = 148,
      MULTINOMIAL = 149,
      GELU = 150,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc

      }
    };
    
    // Checks if the input producer op is supported in this transform. Right now, we
    // only check if it is a tf.IteratorGetNext where resource input is coming from
    // a VarHandle on CPU or a function argument assigned to CPU.
    bool IsSupportedInputOp(
        Operation* op,
        const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td

        because the tensors are loaded asynchronously.
    
      }];
    
      let arguments = (ins
        TFTensorType:$prefix,
        TFTensorType:$tensor_names,
        TFTensorType:$shape_and_slices,
        Variadic<TFTensorType>:$var_handles,
        TypeArrayAttr: $restored_dtypes,
        DenseBoolArrayAttr: $truncate_in_cast
      );
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 20:44:15 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc

        if (variables.count(GetResourceKey(var_handle_op)))
          work_list.push_back(var_handle_op);
      }
    
      // Capture list of ops to be erased by traversing usage starting from
      // the VarHandle ops.
      llvm::SetVector<Operation*> erase_list;
      while (!work_list.empty()) {
        auto* operation = work_list.pop_back_val();
        erase_list.insert(operation);
        for (auto& use : operation->getUses()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
Back to top