- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 93 for attr_val (0.2 sec)
-
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) -
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) -
tensorflow/compiler/mlir/tfr/python/tfr_gen.py
if attr_val.type == types_pb2.DT_FLOAT: return 'f32' elif attr_val.type == types_pb2.DT_INT32: return 'i32' elif attr_val.type == types_pb2.DT_INT64: return 'i64' elif attr_val.type == types_pb2.DT_BOOL: return 'i1' # list if attr_val.HasField('list'): if attr_val.list.f: elt_ty = TFRTypes.F32 values = attr_val.list.f
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K bytes - Viewed (0) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_uniform_quantized_drq.mlir
// Internal functions should be marked as private. They will be inlined and // deleted in `InsertQuantizedFunctionsPass`. // // For Uniform Quantized op case, attributes are generated during quantize // composite pass. Therefore, attr_map is set to an empty string. module { // Currently only 4-d case is supported func.func @quantized_conv2d_fn( %input : tensor<*xf32>, %weight : tensor<*x!tf_type.qint8>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 01 12:06:54 UTC 2022 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_xla_weight_only.mlir
%2 = "tf.Cast"(%identity) {Truncate = false} : (tensor<*xi8>) -> tensor<*xf32> %3 = "tf.Conv3D"(%input, %2) { padding = "VALID", strides = [1, 1, 1, 1, 1], attr_map = "strides:0,padding:1,dilations:2" } : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> func.return %3 : tensor<*xf32> } func.func private @internal_batch_matmul_fn(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 15:43:38 UTC 2023 - 7K bytes - Viewed (0)