| /external/llvm-project/mlir/lib/Dialect/Tosa/Transforms/ |
| D | TosaMakeBroadcastable.cpp | 192 struct ConvertTosaOp<tosa::MulOp> : public OpRewritePattern<tosa::MulOp> {
193 using OpRewritePattern<tosa::MulOp>::OpRewritePattern;
195 LogicalResult matchAndRewrite(tosa::MulOp tosaBinaryOp, in matchAndRewrite()
209 rewriter.replaceOpWithNewOp<tosa::MulOp>(tosaBinaryOp, outputType, in matchAndRewrite()
259 patterns.insert<ConvertTosaOp<tosa::MulOp>>(ctx); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 87 poly = rewriter.create<mhlo::MulOp>(loc, x.getType(), poly, x); in MaterializePolynomialApproximation()
128 Value x_sq = rewriter.create<mhlo::MulOp>(loc, x, x); in MaterializeErfcApproximationF64ForMagnituteGEOne()
137 Value exp_z_mul_poly_p = rewriter.create<mhlo::MulOp>(loc, exp_z, poly_p); in MaterializeErfcApproximationF64ForMagnituteGEOne()
147 Value exp_z_mul_poly_r = rewriter.create<mhlo::MulOp>(loc, exp_z, poly_r); in MaterializeErfcApproximationF64ForMagnituteGEOne()
199 Value x_sq = rewriter.create<mhlo::MulOp>(loc, x, x); in MaterializeErfApproximationF64ForMagnituteLEOne()
202 Value x_mul_poly_t = rewriter.create<mhlo::MulOp>(loc, x, poly_t); in MaterializeErfApproximationF64ForMagnituteLEOne()
286 Value x_sq = rewriter.create<mhlo::MulOp>(loc, x, x); in MaterializeErfcApproximationF32ForMagnitudeGEOne()
300 rewriter.create<mhlo::MulOp>(loc, exp_z, one_div_abs_x); in MaterializeErfcApproximationF32ForMagnitudeGEOne()
310 rewriter.create<mhlo::MulOp>(loc, exp_z_mul_one_div_abs_x, poly); in MaterializeErfcApproximationF32ForMagnitudeGEOne()
346 Value x_sq = rewriter.create<mhlo::MulOp>(loc, x, x); in MaterializeErfApproximationF32ForMagnitudeLEOne()
[all …]
|
| D | lhlo_legalize_to_affine.cc | 146 BinaryOpConverter<lmhlo::MulOp>, in populateLHLOToAffineConversionPattern()
|
| D | unfuse_batch_norm.cc | 163 rewriter.create<mhlo::MulOp>(bn_op.getLoc(), result, broadcast_scale); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 435 struct FuseFullyConnectedAndMul : public OpRewritePattern<TFL::MulOp> {
436 using OpRewritePattern<TFL::MulOp>::OpRewritePattern;
438 LogicalResult matchAndRewrite(TFL::MulOp mul_op, in matchAndRewrite()
488 rewriter.create<TF::MulOp>(mul_op.getLoc(), filter, new_const_val).z(); in matchAndRewrite()
492 rewriter.create<TF::MulOp>(mul_op.getLoc(), bias, constant_val).z(); in matchAndRewrite()
536 struct FuseAffinOpAndMulWithQDQs : public OpRewritePattern<TFL::MulOp> {
537 using OpRewritePattern<TFL::MulOp>::OpRewritePattern;
539 LogicalResult matchAndRewrite(TFL::MulOp mul_op, in matchAndRewrite()
589 rewriter.create<TF::MulOp>(loc, filter, broadcasted_gamma).z(); in matchAndRewrite()
599 auto new_bias = rewriter.create<TF::MulOp>(loc, bias, gamma); in matchAndRewrite()
[all …]
|
| D | load_quantization_recipe.cc | 119 auto s3 = builder->create<MulOp>(loc, int16, cell.getValue().first, in CreateGate()
171 auto forget_cell_state = builder->create<MulOp>( in LoadForLSTMOp()
173 auto input_cell_state = builder->create<MulOp>( in LoadForLSTMOp()
186 auto hidden_state = builder->create<MulOp>( in LoadForLSTMOp()
|
| D | prepare_tf.cc | 881 auto mul_op = rewriter.create<TF::MulOp>( in matchAndRewrite()
1170 ::mlir::TF::MulOp mul_op_1; in matchAndRewrite()
1175 mul_op_1 = rewriter.create<::mlir::TF::MulOp>(odsLoc, tblgen_value_0, in matchAndRewrite()
1197 ::mlir::TF::MulOp mul_op_2; in matchAndRewrite()
1202 mul_op_2 = rewriter.create<::mlir::TF::MulOp>(odsLoc, tblgen_value_0, in matchAndRewrite()
1234 ::mlir::TF::MulOp multiplier; in matchAndRewrite()
1238 multiplier = rewriter.create<::mlir::TF::MulOp>(odsLoc, in matchAndRewrite()
1242 ::mlir::TF::MulOp mul_op_1; in matchAndRewrite()
1246 mul_op_1 = rewriter.create<::mlir::TF::MulOp>(odsLoc, in matchAndRewrite()
1250 ::mlir::TF::MulOp mul_op_2; in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 331 ConstOp::getOperationName(), MulOp::getOperationName(), in ConvertFakeQuantWithMinMaxVarsOp()
394 rewriter.create<MulOp>(op.getLoc(), quant_min_sub, quant_to_float); in matchAndRewrite()
397 rewriter.create<MulOp>(op.getLoc(), quant_max_sub, quant_to_float); in matchAndRewrite()
408 quantized_input = rewriter.create<MulOp>(op.getLoc(), input_ty, in matchAndRewrite()
422 Value output = rewriter.create<MulOp>(op.getLoc(), input_ty, in matchAndRewrite()
537 MulOp::getOperationName(), in LowerLgammaOp()
643 rewriter.create<MulOp>(loc, z_plus_one_half_minus_t_div_log_t, log_t); in matchAndRewrite()
689 rewriter.create<MulOp>(loc, pi, reduced_frac_input); in matchAndRewrite()
813 MulOp::getOperationName(), in LowerSpaceToBatchNDOp()
996 rewriter.create<MulOp>(loc, output_batch, block_shape_splits[i]); in matchAndRewrite()
[all …]
|
| D | decompose_resource_ops.cc | 95 MulOp::getOperationName(), in DecomposeRngReadAndSkipOp()
170 rewriter.create<MulOp>(loc, u64_scalar, step_size, rng_op.delta()); in matchAndRewrite()
|
| /external/deqp/framework/randomshaders/ |
| D | rsgBinaryOps.hpp | 92 class MulOp : public MulBase class
95 MulOp (GeneratorState& state, ConstValueRangeAccess valueRange);
96 virtual ~MulOp (void) {} in ~MulOp()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| D | ShapeToShapeLowering.cpp | 46 Value product = b.create<MulOp>(loc, valueType, body->getArgument(1), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 69 MAP_HLO_TO_LHLO(MulOp);
|
| D | map_chlo_to_hlo_op.h | 73 POPULATE_BCAST(BroadcastMulOp, mhlo::MulOp); in PopulateForBroadcastingBinaryOp()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | Dialect.cpp | 259 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
267 void MulOp::inferShapes() { getResult().setType(getOperand(0).getType()); } in inferShapes()
|
| /external/llvm-project/mlir/examples/toy/Ch5/mlir/ |
| D | Dialect.cpp | 259 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
267 void MulOp::inferShapes() { getResult().setType(getOperand(0).getType()); } in inferShapes()
|
| /external/llvm-project/mlir/examples/toy/Ch4/mlir/ |
| D | Dialect.cpp | 259 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
267 void MulOp::inferShapes() { getResult().setType(getOperand(0).getType()); } in inferShapes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 488 auto op3_mul_op1_op2 = rewriter.create<tosa::MulOp>( in convertMultiplyOp()
496 .create<tosa::MulOp>(op->getLoc(), output_type, input_lhs_val, in convertMultiplyOp()
522 .create<tosa::MulOp>(op->getLoc(), result_type, sub_op.getResult(), in convertSquaredDifferenceOp()
1557 rewriter.create<tosa::MulOp>(op->getLoc(), int32_logits_type, in convertSoftmaxOp()
1597 .create<tosa::MulOp>(op->getLoc(), output_type, op1_exp_in.getResult(), in convertSoftmaxOp()
1657 auto op4_mul_op1_op3 = rewriter.create<tosa::MulOp>( in convertLogSoftmaxOp()
2164 rewriter.create<tosa::MulOp>(op->getLoc(), output_type, lhs_value, in convertFloorDivOp()
2192 rewriter.create<tosa::MulOp>(op->getLoc(), output_type, lhs_value, in convertFloorModOp()
2572 .create<tosa::MulOp>(op->getLoc(), output_type, val.getValue(), in convertReduceMeanOp()
2671 auto op1_mul_in = rewriter.create<tosa::MulOp>( in convertQuantizeOp()
[all …]
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | Dialect.cpp | 311 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
319 void MulOp::inferShapes() { getResult().setType(getOperand(0).getType()); } in inferShapes()
|
| /external/llvm-project/mlir/examples/toy/Ch3/include/toy/ |
| D | Ops.td | 137 def MulOp : Toy_Op<"mul", [NoSideEffect]> {
151 // Allow building a MulOp with from the two input operands.
|
| /external/llvm-project/mlir/examples/toy/Ch2/include/toy/ |
| D | Ops.td | 138 def MulOp : Toy_Op<"mul"> {
152 // Allow building a MulOp with from the two input operands.
|
| /external/llvm-project/mlir/examples/toy/Ch3/mlir/ |
| D | Dialect.cpp | 182 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
|
| /external/llvm-project/mlir/examples/toy/Ch2/mlir/ |
| D | Dialect.cpp | 182 void MulOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
|
| /external/llvm-project/mlir/examples/toy/Ch4/include/toy/ |
| D | Ops.td | 162 def MulOp : Toy_Op<"mul",
177 // Allow building a MulOp with from the two input operands.
|
| /external/llvm-project/mlir/examples/toy/Ch5/include/toy/ |
| D | Ops.td | 162 def MulOp : Toy_Op<"mul",
177 // Allow building a MulOp with from the two input operands.
|
| /external/llvm-project/mlir/examples/toy/Ch6/include/toy/ |
| D | Ops.td | 162 def MulOp : Toy_Op<"mul",
177 // Allow building a MulOp with from the two input operands.
|