| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_trigonometric_to_approximation.cc | 46 if (!op.getType().isF16() && !op.getType().isF32()) return failure(); in matchAndRewrite()
60 if (!arg.getType().isF32()) return failure(); in matchAndRewrite()
66 assert(result.getType().isF32() && "Expect f32 intermediate result."); in matchAndRewrite()
90 assert(input.getType().isF32()); in emitApproximation()
|
| D | chlo_legalize_to_hlo.cc | 271 assert(x.getType().cast<ShapedType>().getElementType().isF32() && in MaterializeErfcApproximationF32ForMagnitudeGEOne()
336 assert(x.getType().cast<ShapedType>().getElementType().isF32() && in MaterializeErfApproximationF32ForMagnitudeLEOne()
356 assert(x.getType().cast<ShapedType>().getElementType().isF32() && in MaterializeErfApproximationF32()
387 assert(x.getType().cast<ShapedType>().getElementType().isF32() && in MaterializeErfcApproximationF32()
450 if (!ty.isF64() && !ty.isF32() && !ty.isF16()) return failure(); in matchAndRewrite()
475 if (!ty.isF64() && !ty.isF32() && !ty.isF16()) return failure(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils.cc | 178 if (!image_type || !image_type.getElementType().isF32() || in VerifySignature()
185 if (!flow_type || !flow_type.getElementType().isF32() || in VerifySignature()
192 if (!output_type || !output_type.getElementType().isF32() || in VerifySignature()
|
| D | constant_utils.cc | 43 } else if (element_type.isF32()) { in CreateConstOpWithSingleValue()
48 if (etype.isF32()) { in CreateConstOpWithSingleValue()
|
| D | validators.h | 72 return tensorType.getElementType().isF32(); in TFTypeIsFloat32Tensor()
|
| /external/llvm-project/mlir/test/mlir-tblgen/ |
| D | predicate.td | 11 def I32OrF32 : Type<CPred<"$_self.isInteger(32) || $_self.isF32()">,
21 // CHECK: if (!((v.getType().isInteger(32) || v.getType().isF32())))
96 …orType>())) && (((v.getType().cast<::mlir::ShapedType>().getElementType().isF32())) || ((v.getType…
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_memref_prints.cc | 64 if (!element_type.isF32() && !element_type.isF64() && in EmitPrint()
84 if (element_type.isF32()) { in EmitPrint()
|
| /external/llvm-project/mlir/lib/Conversion/AVX512ToLLVM/ |
| D | ConvertAVX512ToLLVM.cpp | 85 if (!getSrcVectorElementType(cast<MaskRndScaleOp>(op)).isF32()) in matchAndRewrite()
119 if (!getSrcVectorElementType(cast<MaskScaleFOp>(op)).isF32()) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 204 if (!conv.T().isF32() && !conv.T().isF64()) { in AreFuseCompatible()
225 if (!matmul.T().isF32() && !matmul.T().isBF16()) { in AreFuseCompatible()
|
| D | optimize.td | 44 … CPred<"$_self.cast<ElementsAttr>().getType().getElementType().isF32()">, "float constant tensor">;
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_type.cc | 97 } else if (type.isF32()) { in ConvertScalarTypeToDataType()
129 if (etype.isF32()) { in ConvertScalarTypeToDataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 50 } else if (type.isF32()) { in TypeToPrimitiveType()
56 if (element_ty.isF32()) { in TypeToPrimitiveType()
|
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestPatterns.cpp | 363 if (op->getNumOperands() != 1 || !op->getOperand(0).getType().isF32()) in matchAndRewrite()
401 if (!Type(*op->result_type_begin()).isF32()) in matchAndRewrite()
517 if (t.isF32()) { in convertType()
582 [](Type type) { return type.isF32(); }); in runOnOperation()
793 if (type.isBF16() || type.isF16() || type.isF32()) in runOnOperation()
834 return (*op.operand_type_begin()).isF64() && op.getType().isF32(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | hlo_utils.h | 51 if (complex_element_ty.isF32()) in getSplat()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.cc | 100 } else if (type.isF32()) { in ConvertDerivedTypeAttrForOptionWriter()
105 if (complex_type.getElementType().isF32()) { in ConvertDerivedTypeAttrForOptionWriter()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Types.cpp | 27 bool Type::isF32() const { return isa<Float32Type>(); } in isF32() function in Type
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | while_loop_outline.cc | 79 if (elemType.isBF16() || elemType.isF32()) return true; in IsCompatibleTypeWithTFLCastOp()
88 elemType.cast<ComplexType>().getElementType().isF32()) in IsCompatibleTypeWithTFLCastOp()
|
| D | quantize_patterns.td | 25 CPred<"$_self.cast<ElementsAttr>().getType().getElementType().isF32()">, "float constant tensor">;
|
| D | dense_to_sparse.cc | 118 if (type.getElementType().isF32()) { in CalculateBlockSparsity()
345 if (type.getElementType().isF32()) { in runOnFunction()
|
| D | default_quant_params.cc | 155 if (!tensor_type.getElementType().isF32()) return; in AddToWorkListIfUnquantized()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | hlo_utils.cc | 68 if (complex_element_ty.isF32()) { in GetScalarOfType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_types.cc | 108 } else if (type.isF32()) { in get()
116 if (etype.isF32()) { in get()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 191 if (element_type.isF32() || IsQI8Type(element_type) || in VerifyAddOpShapeConstraints()
212 if (element_type.isF32() || IsI32Type(element_type) || in VerifySubOpShapeConstraints()
250 element_type.isF32()) { in VerifyMulOpShapeConstraints()
327 return shaped_type.getElementType().isF32(); in IsF32ShapedType()
866 if (!input_type.getElementType().isF32() || in fold()
867 !weights_type.getElementType().isF32() || in fold()
868 !output_type.getElementType().isF32() || in fold()
869 (has_bias && !bias_type.getElementType().isF32())) { in fold()
1130 if (getType().cast<ShapedType>().getElementType().isF32()) { in fold()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantTypes.h | 182 using Type::isF32;
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Types.h | 119 bool isF32() const;
|