Searched refs:fp_type (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 67 FloatType fp_type, Value variance, in MaterializeEpsilon() argument
71 if (epsilon_attr.getType() != fp_type) { in MaterializeEpsilon()
76 fp_type.getFloatSemantics(), APFloat::rmNearestTiesToEven, &loses_info); in MaterializeEpsilon()
86 epsilon_attr = b.getFloatAttr(fp_type, epsilon_float); in MaterializeEpsilon()
89 auto scalar_type = RankedTensorType::get({}, fp_type); in MaterializeEpsilon()
122 auto fp_type = variance_type.getElementType().dyn_cast<FloatType>(); in matchAndRewrite() local
123 if (!fp_type) { in matchAndRewrite()
131 MaterializeEpsilon(bn_op.getOperation(), bn_op.epsilonAttr(), fp_type, in matchAndRewrite()
|
| /external/skqp/src/sksl/ |
| D | SkSLContext.h | 199 , fFragmentProcessor_Type(fp_type(fInt_Type.get(), fBool_Type.get()))
389 static std::unique_ptr<Type> fp_type(const Type* intType, const Type* boolType) { in fp_type() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 1105 PrimitiveType fp_type = PRIMITIVE_TYPE_INVALID; in CheckMixedPrecisionOperands() local
1113 if (fp_type == PRIMITIVE_TYPE_INVALID) { in CheckMixedPrecisionOperands()
1114 fp_type = subshape.element_type(); in CheckMixedPrecisionOperands()
1115 } else if (fp_type != subshape.element_type()) { in CheckMixedPrecisionOperands()
|