- Sort Score
- Result 10 results
- Languages All
Results 101 - 107 of 107 for Dtype (0.03 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
LogicalResult ConstantFoldMul(mhlo::MulOp op, PatternRewriter &rewriter) { ShapedType type = mlir::dyn_cast<ShapedType>(op.getType()); Type etype = type.getElementType(); Attribute result = {}; if (mlir::isa<FloatType>(etype)) { result = BinaryFolder<mhlo::MulOp, FloatType, APFloat, std::multiplies<APFloat>>( &op); } else if (mlir::isa<IntegerType>(etype)) { result =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.cc
tflite::TensorType ConvertTypeToTensorType(mlir::Type type) { if (type.isF16()) { return tflite::TensorType_FLOAT16; } else if (type.isBF16()) { return tflite::TensorType_BFLOAT16; } else if (type.isF32()) { return tflite::TensorType_FLOAT32; } else if (type.isF64()) { return tflite::TensorType_FLOAT64; } else if (mlir::isa<mlir::TF::StringType>(type)) { return tflite::TensorType_STRING;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
return success(); } static LogicalResult FoldDivOp(stablehlo::DivOp op, PatternRewriter& rewriter) { auto etype = op.getType().getElementType(); if (etype.isa<FloatType>()) { return FoldDivOpInternal<APFloat>(op, rewriter); } if (etype.isa<IntegerType>()) { return FoldDivOpInternal<APInt>(op, rewriter); } return failure(); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
rewriter.replaceOpWithNewOp<mhlo::ReturnOp>(op, adaptor.getOperands()); return success(); } }; // Returns a new tensor type from the given type with element type updated to // the given type. TensorType UpdateElementTypeTo(Type ty, Type element_ty) { auto ranked_ty = mlir::dyn_cast<RankedTensorType>(ty); if (!ranked_ty) { return UnrankedTensorType::get(element_ty); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
auto attr_key = attr.first; auto attr_value = attr.second; if (attr_key == attrKey && attr_value.value_case() == AttrValue::kType && attr_value.type() == DT_RESOURCE) { return true; break; } } } } } return false; } bool IsNonReplicatedGraph(const Graph& graph,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
attr.default_value().value_case() == AttrValue::kType) { type = DataType_Name(attr.default_value().type()); } else if (attr.has_allowed_values() && attr.allowed_values().value_case() == AttrValue::kList && !attr.allowed_values().list().type().empty()) { type = DataType_Name(attr.allowed_values().list().type(0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc
type, builder.getFloatAttr(element_ty, unique_index)); if (auto qtype = mlir::dyn_cast<QuantizedType>(element_ty)) { mlir::RankedTensorType new_type = tensorflow::GetTypeFromTFTensorShape( type.getShape(), qtype.getStorageType()); return DenseElementsAttr::get( new_type, builder.getIntegerAttr(qtype.getStorageType(), unique_index)); } llvm_unreachable("unhandled element type"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0)