Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 52 for attr_val (0.19 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

          `attr_name` is given, check if the `attr_val` matches with the attribute
          value of the op.
        """
        # Check the main graph
        if self._contains_op_with_name_and_attribute(
            nodes=graphdef.node,
            op_name=op_name,
            attr_name=attr_name,
            attr_val=attr_val,
            node_name=node_name,
        ):
          return True
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

          }
          TF_RETURN_IF_ERROR(
              ConvertAttribute(elt_type, remove_ref_type, &attr_val));
          list->add_type(attr_val.type());
        } else if (auto attr = mlir::dyn_cast<mlir::TF::ShapeAttr>(a)) {
          AttrValue attr_val;
          TF_RETURN_IF_ERROR(ConvertAttribute(attr, &attr_val));
          *list->add_shape() = attr_val.shape();
        } else if (auto attr = mlir::dyn_cast<mlir::ArrayAttr>(a)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

        if (attribute_getter_map.find(attr_name.str()) !=
            attribute_getter_map.end()) {
          auto attr_val =
              (attribute_getter_map[attr_name.str()])(rewriter, identifier_to_attr);
          attrs.push_back(rewriter.getNamedAttr(attr_name, attr_val));
        }
      }
    
      auto feature_group_cnt_attr = llvm::StringRef("feature_group_count");
      int feature_group_cnt = 1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

        # Convolution ouside the while op is quantized.
        self.assertTrue(
            self._contains_op(
                output_graphdef,
                op_name='XlaConvV2',
                attr_name='RhsT',
                attr_val=attr_value_pb2.AttrValue(type=types_pb2.DT_INT8),
            )
        )
        # TODO: b/294783597 - [Converter][TF-Quantizer] Support quantization for the
        # ops in the while op body for both SRQ and WO
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

      tensorflow::AttrValue attr_value;
      if (!attr_value.ParseFromArray(proto, proto_len)) {
        status->status = InvalidArgument("Unparseable AttrValue proto");
        return;
      }
    
      if (strcmp(attr_name, tensorflow::kColocationAttrName) == 0) {
        if (attr_value.value_case() != tensorflow::AttrValue::kList &&
            attr_value.value_case() != tensorflow::AttrValue::VALUE_NOT_SET) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/cc_op_gen_util.cc

    }
    
    string PrintAttrValue(const string& op, const AttrValue& attr_value) {
      switch (attr_value.value_case()) {
        case AttrValue::kS:
          return PrintString(attr_value.s());
        case AttrValue::kI:
          return strings::StrCat(attr_value.i());
        case AttrValue::kF: {
          const float f = attr_value.f();
          if (std::isinf(f)) {
            return strings::StrCat(f < 0.0f ? "-" : "+",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc

    string AttrView::DefaultValue() const {
      const AttrValue &attr_value = attr_.default_value();
      switch (attr_value.value_case()) {
        case AttrValue::VALUE_NOT_SET:
          return "";
        case AttrValue::kType:
          return DataType_Name(attr_value.type());
        case AttrValue::kS:
          return "\"" + attr_value.s() + "\"";
        case AttrValue::kI:
          return std::to_string(attr_value.i());
        case AttrValue::kF:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/xla_cluster_util.cc

      for (const auto& name_attr_pair : n.attrs()) {
        const AttrValue& attr_value = name_attr_pair.second;
        if (attr_value.value_case() == AttrValue::kFunc) {
          result.push_back(attr_value.func());
        } else if (attr_value.value_case() == AttrValue::kList) {
          result.insert(result.end(), attr_value.list().func().begin(),
                        attr_value.list().func().end());
        }
      }
    
      return result;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions.mlir

    // CHECK-NEXT: %[[matmul:.*]] = "tf.MatMul"(%arg0, %arg1)
    // CHECK-SAME: attr_map = "0:transpose_a,1:transpose_b"
    // CHECK-NEXT: tf.BiasAdd
    // CHECK-NEXT: tf.Relu6
    // CHECK-NEXT: return
    
    // CHECK-LABEL: private @composite_matmul_with_bias_and_relu_fn_1
    // CHECK-NEXT: tf.MatMul"(%arg0, %arg1)
    // CHECK-SAME: attr_map = "0:transpose_a,1:transpose_b"
    // CHECK-NEXT: tf.BiasAdd
    // CHECK-NEXT: tf.Relu
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 26.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/tests/add_dump_tensor_op.mlir

      func.func private @composite_conv2d_with_bias_and_relu6_fn_2(%arg0: tensor<1x2x2x3xf32>, %arg1: tensor<2x2x3x2xf32>, %arg2: tensor<2xf32>) -> tensor<*xf32> attributes {tf_quant.composite_function} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 22:55:22 UTC 2024
    - 37.9K bytes
    - Viewed (0)
Back to top