- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 140 for getElementDtype (0.27 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
return list.emitOpError("unknown tensor list element shape"); } } if (failed(cutil::CreateInitBufferValue( element_shape, list.getMaxNumElements(), list, list.getElementDtype(), builder, &buffer))) { return failure(); } Value size = cutil::GetR1Const({0LL}, builder, list.getLoc()); list.getHandle().replaceAllUsesWith(buffer);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
return UnrankedTensorType::get( variant_ty.getSubtypes()[0].getElementType()); } Type value_type = value.getType(); Type element_type; variant_ty = value_type.dyn_cast<TF::VariantType>(); if (variant_ty && !variant_ty.getSubtypes().empty()) { element_type = variant_ty.getSubtypes()[0].getElementType(); } else { element_type = getElementTypeOrSelf(value_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
if (!RankEquals(value_type, 1) || !mlir::isa<StringType>(value_type.getElementType())) { return func.emitError() << "1st output should be string tensor"; } if (func.getNumResults() > 1) { auto offset_type = GetResultType(func, 1); if (!RankEquals(offset_type, 1) || !offset_type.getElementType().isInteger(64)) { return func.emitError() << "2nd output should be int64 tensor"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
allTypesQuantizedOrInt &= (mlir::isa<quant::QuantizedType>(type.getElementType()) || mlir::isa<IntegerType>(type.getElementType())); } for (auto result : op->getResults()) { ShapedType type = mlir::cast<ShapedType>(result.getType()); allTypesFp &= !mlir::isa<quant::QuantizedType>(type.getElementType()); allTypesQuantizedOrInt &=
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/flatbuffer_operator.h
// we're just returning empty vector template <> inline std::vector<bool> GetVector(DenseElementsAttr elements) { auto type = elements.getType(); auto elemType = type.getElementType(); if (elemType.isSignlessInteger(1)) { auto vec = llvm::to_vector( llvm::map_range(elements.getValues<bool>(), [&](bool value) -> uint8_t { return value ? 1 : 0; }));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 21:00:09 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
int64_t num_units = output_type.getDimSize(output_type.getRank() - 1); auto bias_type = mlir::RankedTensorType::get({num_units}, output_type.getElementType()); mlir::DenseElementsAttr bias_attr; if (output_type.getElementType().isF32()) { float val = 0.0; bias_attr = mlir::DenseFPElementsAttr::get(bias_type, val); } else { // TODO(renjieliu): Refactor this and share the logic with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/attribute_utils.cc
namespace mlir { namespace TFL { FloatAttr ExtractSingleElementAsFloat(ElementsAttr attr) { if (attr.getShapedType().getNumElements() != 1 || !mlir::isa<FloatType>(attr.getShapedType().getElementType())) { return {}; } return attr.getSplatValue<FloatAttr>(); } FloatAttr GetSingleElementAsFloatOrSelf(Attribute attr) { if (auto m = mlir::dyn_cast_or_null<ElementsAttr>(attr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/QuantOps.cc
return quantizedType.isCompatibleExpressedType(expressed); if (auto tensorType = mlir::dyn_cast<TensorType>(expressed)) return spec == tensorType.getElementType(); if (auto vectorType = mlir::dyn_cast<VectorType>(expressed)) return spec == vectorType.getElementType(); } return false; } LogicalResult QuantizeRegionOp::verify() { // There are specifications for both inputs and outputs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.cc
return quantizedType.isCompatibleExpressedType(expressed); if (auto tensorType = mlir::dyn_cast<TensorType>(expressed)) return spec == tensorType.getElementType(); if (auto vectorType = mlir::dyn_cast<VectorType>(expressed)) return spec == vectorType.getElementType(); } return false; } LogicalResult QuantizeRegionOp::verify() { // There are specifications for both inputs and outputs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0)