- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 53 for qtype_attr (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype); value->set_type(dtype); return absl::OkStatus(); } Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type, AttrValue* value) { return ConvertAttribute(type.getValue(), remove_ref_type, value); } Status ConvertAttribute(const mlir::UnitAttr& attr, AttrValue* value) {
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/ir/tfr_ops.td
def TFR_allowedArgType : Type<Or<[ TFR_AllTensorTypes.predicate, TFR_AllAttrTypes.predicate]>, "allowed tfr.call operand types">; def TFR_allowedConstValues : Attr<Or<[ FlatSymbolRefAttr.predicate, TypeAttr.predicate, StrAttr.predicate, ArrayAttr.predicate]>, "allowed tfr.constant value"> { let storageType = "Attribute"; let returnType = "Attribute"; let convertFromStorage = "$_self";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
Type resultTy = op->getOpResult(i).getType(); if (QuantizedType::getQuantizedElementType(resultTy)) { replaceResults.push_back(rewriter.create<TFL::QuantizeOp>( loc, resultTy, result, TypeAttr::get(resultTy))); continue; } replaceResults.push_back(result); } rewriter.replaceOp(op, replaceResults); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
output_shape, mlir::cast<quant::UniformQuantizedType>(output_type.getElementType()) .getStorageType()); rewriter.replaceOpWithNewOp<QConstOp>( op, TypeAttr::get(result_type), DenseIntElementsAttr::get(values_type, new_values)); return success(); } }; // Fold constant quantized Reshape ops. struct FoldReshapeOp : public OpRewritePattern<ReshapeOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
} return string(name); } void InferArgAttributes(const OpDef::ArgDef& arg, std::unordered_map<string, string>* inferred_attrs) { if (!arg.type_attr().empty()) { gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name()); } else if (!arg.type_list_attr().empty()) { gtl::InsertIfNotPresent(inferred_attrs, arg.type_list_attr(), arg.name()); } if (!arg.number_attr().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc
if (IsQI8Type(result_type) || IsQUI8Type(result_type)) { builder->setInsertionPoint(op); TFL::QuantizeOp quant_op = builder->create<TFL::QuantizeOp>( op->getLoc(), result_type, new_result, TypeAttr::get(result_type)); new_result = quant_op.getResult(); } // Rewire the outputs. result.replaceAllUsesWith(new_result); } // Remove the old op. op->erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
// the casting, the quantization dimension of the result type needs to be set // this new `axis` value. TypeAttr CastQuantizedTypeAttrFromExpressedType(Builder builder, TypeAttr source, Type target, int axis); // Quantizes the elements in the attribute `real_value` by the quantization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
builder.create<tf_saved_model::GlobalTensorOp>( NameLoc::get(builder.getStringAttr(name.str())), builder.getStringAttr(name), tensor_attr, TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr()); } return success(); } } // namespace LogicalResult LiftVariables(ModuleOp module, Session* session) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
values.begin(), values.end(), [](const APInt& element_value) { return element_value.isNegative(); }); } // Utility function to get the offset between two dense attribute values. inline TypedAttr GetOffSet(Attribute begin, Attribute end) { auto begin_values = begin.cast<DenseElementsAttr>().getValues<int32_t>(); auto end_values = end.cast<DenseElementsAttr>().getValues<int32_t>(); SmallVector<int32_t> offsets;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
return f.isExactlyValue(value); }); } // Returns true if `value` is compile-time constant and its splat value equals // to `raw_value`. template <typename T> bool IsConstantValueOf(mlir::TypedAttr value, T raw_value) { auto element_type = mlir::cast<ShapedType>(value.getType()).getElementType(); if (mlir::isa<FloatType>(element_type)) { return FloatValueEquals(value, raw_value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)