- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for TypeAttr (0.13 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
/*asymmetric_quantize_inputs=*/mlir::BoolAttr(), /*input_to_input_intermediate=*/mlir::TypeAttr(), /*input_to_forget_intermediate=*/mlir::TypeAttr(), /*input_to_cell_intermediate=*/mlir::TypeAttr(), /*input_to_output_intermediate=*/mlir::TypeAttr(), /*effective_hidden_scale_intermediate=*/mlir::TypeAttr()); // Cast the static shaped lstm result to FuncOp's signature -
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/quantization_context.cc
input_specs.push_back(original_input_specs[i]); } else if (requantize.pos == RequantizeState::ON_OUTPUT) { input_specs.push_back(TypeAttr::get(requantize.params)); } else { input_specs.push_back(TypeAttr::get(state.params)); } } op->setAttr("input_specs", ArrayAttr::get(context, input_specs)); llvm::SmallVector<Attribute, 4> output_specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 01:38:03 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
// index. template <typename LstmOp> inline QuantizedType GetIntermediateElementType(LstmOp op, int tensor_index) { if (tensor_index < 0 || tensor_index > 4) return nullptr; TypeAttr attr = op->template getAttrOfType<TypeAttr>( intermediate_attributes[tensor_index]); if (!attr) { return nullptr; } return QuantizedType::getQuantizedElementType(attr.getValue()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
if (input_types.size() <= 1) return; Type target_input_type = mlir::cast<TypeAttr>(input_types[0]).getValue(); auto result_type = UnrankedTensorType::get(target_input_type); for (auto i = 1; i < input_types.size(); ++i) { Type current_input_type = mlir::cast<TypeAttr>(input_types[i]).getValue(); if (current_input_type != target_input_type) { input_values[i] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
legacy_float_scale, use_fake_quant_num_bits); if (!final_type) return {}; return TypeAttr::get(final_type); } TypeAttr CastQuantizedTypeAttrFromExpressedType(const Builder builder, const TypeAttr source, const Type target, const int axis) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
auto c_false = rewriter.create<arith::ConstantOp>(loc, rewriter.getBoolAttr(false)); TypeAttr f32_attr = TypeAttr::get(rewriter.getF32Type()); TFRAttrType output_type = TFRAttrType::get(rewriter.getContext()); auto constant_f32_op = rewriter.create<ConstOp>(loc, output_type, f32_attr); TypeAttr i32_attr = TypeAttr::get(rewriter.getI32Type()); auto constant_i32_op = rewriter.create<ConstOp>(loc, output_type, i32_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
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/tfr/passes/decompose.cc
} attribute = TypeAttr::get(type); } Value attr_cst; // Wrap these special attributes as a special TFR constant, so the SSA // value has a valid type to be used as TFR function argument. These // attributes are not expected to be manipulated by the lowering passes. if (mlir::isa<TypeAttr>(attribute) || mlir::isa<ArrayAttr>(attribute) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0)