- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for isF32 (0.21 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
// Checks if the param passed is a F32 ElementsAttr. def F32ElementsAttr : ElementsAttrBase< CPred<"$_self.isa<ElementsAttr>() && $_self.cast<ElementsAttr>().getShapedType().getElementType().isF32()">, "32 bit float constant tensor">; // Checks if the param passed is a float ElementsAttr. def FloatElementsAttr : ElementsAttrBase<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
GetFilterConstantOp(filter_value); auto filter_value_attr = mlir::cast<DenseElementsAttr>(filter_constant_op.getValue()); if (filter_value_attr.getElementType().isF32()) { // This is i8 values disguised as f32 (due to the upcast trick). Simply // cast them to i8. filter_value_attr = mlir::cast<DenseFPElementsAttr>(filter_value_attr)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
.isF32()) { return failure(); } MLIRContext *context = rewriter.getContext(); llvm::SmallVector<Value, 2> operands{op.getA(), op.getB()}; for (Value &operand : operands) { TensorType tensor_type = mlir::cast<TensorType>(operand.getType()); Type element_type = tensor_type.getElementType(); if (element_type.isF32()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
return base_function_name.concat("_fn").str(); } bool ContainsFloatResultType(ArrayRef<Type> result_types) { for (auto current_type : result_types) { if (mlir::dyn_cast<TensorType>(current_type).getElementType().isF32()) return true; } return false; } // Unwraps quantization parameters of PartitionedCall ops with quantized // input/outputs that are created from QuantizePass. class QuantizeFunctionPattern
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
if (!epsilon) epsilon = rewriter.getFloatAttr(rewriter.getF32Type(), 0.0001f); if (!(((mlir::isa<::mlir::FloatAttr>(epsilon))) && ((mlir::cast<::mlir::FloatAttr>(epsilon).getType().isF32())))) { return rewriter.notifyMatchFailure( fused_batch_norm_op, [&](::mlir::Diagnostic &diag) { diag << "op 'tf.FusedBatchNormV3' attribute 'epsilon' failed to "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
LogicalResult matchAndRewrite( OpT op, typename OpT::Adaptor adaptor, ConversionPatternRewriter &rewriter) const override { Type dtype = op.getElementDtype(); if (!(dtype.isF16() || dtype.isF32() || dtype.isF64() || dtype.isInteger(1) || dtype.isInteger(8) || dtype.isInteger(16) || dtype.isInteger(32) || dtype.isInteger(64))) { const char *error_info =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)