/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_ops_n_z.cc | 151 if (matchPattern(op.depth(), m_Constant(&depth_attr))) { in verify() 178 if (matchPattern(depth, m_Constant(&depth_attr)) && in InferOneHotOpType() 626 if (matchPattern(start, m_Constant(&start_val)) && in build() 627 matchPattern(limit, m_Constant(&limit_val)) && in build() 628 matchPattern(delta, m_Constant(&delta_val))) { in build() 761 if (!matchPattern(shape, m_Constant(&shape_attr))) { in GetReshapeOutputType() 1276 if (matchPattern(op.begin(), m_Constant(&begin_indices))) { in verify() 1279 matchPattern(op.size(), m_Constant(&slice_sizes)); in verify() 1312 if (matchPattern(op.size(), m_Constant(&slice_sizes))) { in verify() 1424 if (matchPattern(op.block_shape(), m_Constant(&block_shape_attr))) { in verify() [all …]
|
H A D | tf_ops_a_m.cc | 189 if (matchPattern(op.condition(), m_Constant(&cst))) { in matchAndRewrite() 366 if (matchPattern(op.crops(), m_Constant(&crops_attr))) { in verify() 625 if (!matchPattern(input, m_Constant(&cst_attr))) return {}; in fold() 642 if (!matchPattern(op.s0(), m_Constant(&s0))) return false; in ExtractInputConstShape() 643 if (!matchPattern(op.s1(), m_Constant(&s1))) return false; in ExtractInputConstShape() 791 if (!matchPattern(op.branch_index(), m_Constant(&branch))) return failure(); in matchAndRewrite() 1147 if (!matchPattern(op.axis(), m_Constant(&axis_attr))) return failure(); in matchAndRewrite() 1378 if (matchPattern(op.axis(), m_Constant(&axis_attr))) { in Verify() 2213 if (matchPattern(index, m_Constant(&index_attr))) { in verify() 2503 if (!matchPattern(dim, m_Constant(&dim_attr)) || in InferExpandDimsOpType() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
H A D | optimize.cc | 513 if (!matchPattern(value, m_Constant(&vals))) return false; in AllValuesAreZero() 561 if (!matchPattern(constant_val, m_Constant(&added_value))) return failure(); in matchAndRewrite() 589 if (!is_none_bias && !matchPattern(bias, m_Constant(&bias_value))) in matchAndRewrite() 826 if (!matchPattern(constant_val, m_Constant(&cst))) return failure(); in matchAndRewrite() 835 if (!matchPattern(filter, m_Constant(&cst_tmp))) return failure(); in matchAndRewrite() 837 !matchPattern(bias, m_Constant(&cst_tmp))) in matchAndRewrite() 927 if (!matchPattern(gamma, m_Constant(&gamma_cst))) return failure(); in matchAndRewrite() 947 if (!matchPattern(filter, m_Constant(&cst_tmp))) return failure(); in matchAndRewrite() 949 !matchPattern(bias, m_Constant(&cst_tmp))) in matchAndRewrite() 1017 if (!matchPattern(binary_op->getOperand(1), m_Constant(&cst))) in matchAndRewrite() [all …]
|
H A D | dilated_conv.h | 263 if (!matchPattern(pad_op.paddings(), m_Constant(&pad_attr))) { in matchAndRewrite() 353 if (!matchPattern(stb_paddings, m_Constant(&stb_paddings_attr)) || in matchAndRewrite() 354 !matchPattern(bts_crops, m_Constant(&bts_crops_attr))) { in matchAndRewrite() 448 if (!matchPattern(stb_block_shape, m_Constant(&stb_bs_attr)) || in ExtractDilationsAttrFromBlockShape() 449 !matchPattern(bts_block_shape, m_Constant(&bts_bs_attr))) { in ExtractDilationsAttrFromBlockShape()
|
H A D | legalize_tf.cc | 212 if (!matchPattern(tf_concat_op.axis(), m_Constant(&axis))) return failure(); in matchAndRewrite() 430 if (!matchPattern(tf_matrix_diag_v2_or_v3_op.k(), m_Constant(&k))) in ConvertTFMatrixDiagV2orV3() 437 m_Constant(&num_rows))) in ConvertTFMatrixDiagV2orV3() 444 m_Constant(&num_cols))) in ConvertTFMatrixDiagV2orV3() 454 if (!matchPattern(padding_value, m_Constant(&padding_attr)) || in ConvertTFMatrixDiagV2orV3() 461 if (!matchPattern(padding_value, m_Constant(&padding_attr)) || in ConvertTFMatrixDiagV2orV3()
|
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/InstCombine/ |
H A D | InstCombineMulDivRem.cpp | 199 if (match(&I, m_Mul(m_Shl(m_Value(NewOp), m_Constant(C2)), in visitMul() 200 m_Constant(C1))) && in visitMul() 214 if (match(&I, m_Mul(m_Value(NewOp), m_Constant(C1)))) { in visitMul() 268 if (match(Op0, m_OneUse(m_Add(m_Value(X), m_Constant(C1))))) { in visitMul() 280 if (match(Op0, m_Neg(m_Value(X))) && match(Op1, m_Constant(Op1C))) in visitMul() 423 if (match(Op0, m_FNeg(m_Value(X))) && match(Op1, m_Constant(C))) in visitFMul() 437 if (match(Op1, m_Constant(C)) && C->isFiniteNonZeroFP()) { in visitFMul() 439 if (match(Op0, m_OneUse(m_FDiv(m_Constant(C1), m_Value(X))))) { in visitFMul() 445 if (match(Op0, m_FDiv(m_Value(X), m_Constant(C1)))) { in visitFMul() 461 if (match(Op0, m_OneUse(m_FAdd(m_Value(X), m_Constant(C1))))) { in visitFMul() [all …]
|
H A D | InstCombineAddSub.cpp | 831 if (!match(Op1, m_Constant(Op1C))) in foldNoWrapAdd() 849 if (match(Op0, m_OneUse(m_SExt(m_NSWAdd(m_Value(X), m_Constant(NarrowC)))))) { in foldNoWrapAdd() 856 if (match(Op0, m_OneUse(m_ZExt(m_NUWAdd(m_Value(X), m_Constant(NarrowC)))))) { in foldNoWrapAdd() 869 if (!match(Op1, m_Constant(Op1C))) in foldAddWithConstant() 879 if (match(Op0, m_Sub(m_Constant(Op00C), m_Value(X)))) in foldAddWithConstant() 1142 m_ZExtOrSelf(m_Sub(m_Constant(C), m_ZExtOrSelf(m_Value(NBits))))) || in canonicalizeCondSignextOfHighBitExtractToSignextHighBitExtract() 1186 m_Shl(m_Constant(SignExtendingValueBaseConstant), in canonicalizeCondSignextOfHighBitExtractToSignextHighBitExtract() 1771 if (match(Op1, m_Sub(m_Constant(C2), m_Value(X)))) in visitSub() 1775 if (match(Op1, m_Add(m_Value(X), m_Constant(C2)))) in visitSub() 1904 if (match(Op0, m_OneUse(m_And(m_Specific(Op1), m_Constant(C))))) { in visitSub() [all …]
|
H A D | InstCombineVectorOps.cpp | 918 match(InsElt2.getOperand(1), m_Constant(ScalarC)) && in hoistInsEltConst() 940 if (!match(Shuf->getOperand(1), m_Constant(ShufConstVec)) || in foldConstantInsEltIntoShuffle() 941 !match(InsElt.getOperand(1), m_Constant(InsEltScalar)) || in foldConstantInsEltIntoShuffle() 990 !match(InsElt.getOperand(1), m_Constant(Val[0])) || in foldConstantInsEltIntoShuffle() 992 !match(IEI->getOperand(1), m_Constant(Val[1]))) in foldConstantInsEltIntoShuffle() 1436 if (match(BO1, m_Constant(C))) { in getAlternateBinop() 1463 if (match(Op0, m_BinOp(m_Specific(Op1), m_Constant(C)))) in foldSelectShuffleWith1Binop() 1465 else if (match(Op1, m_BinOp(m_Specific(Op0), m_Constant(C)))) in foldSelectShuffleWith1Binop() 1571 if (match(B0, m_BinOp(m_Value(X), m_Constant(C0))) && in foldSelectShuffle() 1572 match(B1, m_BinOp(m_Value(Y), m_Constant(C1)))) in foldSelectShuffle() [all …]
|
H A D | InstCombineShifts.cpp | 396 if (match(Op0, m_Constant()) && match(Op1, m_Add(m_Value(A), m_Constant(C)))) in commonShiftTransforms() 1020 if (match(Op1, m_Constant(C1))) { in visitShl() 1024 if (match(Op0, m_OneUse(m_Shl(m_Constant(C2), m_Value(X))))) in visitShl() 1028 if (match(Op0, m_Mul(m_Value(X), m_Constant(C2)))) in visitShl() 1197 m_ZExtOrSelf(m_Sub(m_Constant(C1), in foldVariableSignZeroExtensionOfVariableHighBitExtract() 1199 m_ZExtOrSelf(m_Sub(m_Constant(C2), in foldVariableSignZeroExtensionOfVariableHighBitExtract() 1218 m_Sub(m_Constant(C0), m_ZExtOrSelf(m_Specific(NBits))))) || in foldVariableSignZeroExtensionOfVariableHighBitExtract()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
H A D | legalize_tf.cc | 479 if (!matchPattern(tf_argmax_op.dimension(), m_Constant(&axis_elems))) in matchAndRewrite() 644 if (!matchPattern(tf_concatv2_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite() 781 if (!matchPattern(tf_fill_op.dims(), m_Constant(&dims_elems))) in matchAndRewrite() 791 if (!matchPattern(tf_fill_op.value(), m_Constant(&value_elem))) in matchAndRewrite() 1010 m_Constant(&output_shape_elems))) { in matchAndRewrite() 1046 if (!matchPattern(tf_all_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() 1068 if (!matchPattern(tf_any_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() 1090 if (!matchPattern(tf_max_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() 1112 if (!matchPattern(tf_min_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() 1134 if (!matchPattern(tf_mean_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() [all …]
|
H A D | legalize_tfl.cc | 1145 m_Constant(&output_shape_elems))) { in matchAndRewrite() 1642 if (!matchPattern(tfl_reshape_op.shape(), m_Constant(&shape_attr))) { in matchAndRewrite() 1770 if (!matchPattern(tfl_fill_op.dims(), m_Constant(&dims_elems))) in matchAndRewrite() 1780 if (!matchPattern(tfl_fill_op.input(), m_Constant(&value_elem))) in matchAndRewrite() 1813 if (!matchPattern(tfl_any_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite() 1835 if (!matchPattern(tfl_max_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite() 1857 if (!matchPattern(tfl_min_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite() 1879 if (!matchPattern(tfl_prod_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite() 1901 if (!matchPattern(tfl_mean_op.axis(), m_Constant(&axes_elems))) in matchAndRewrite() 1923 if (!matchPattern(tfl_sum_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite() [all …]
|
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/InstCombine/ |
H A D | InstCombineMulDivRem.cpp | 223 if (match(&I, m_Mul(m_Shl(m_Value(NewOp), m_Constant(C2)), in visitMul() 224 m_Constant(C1))) && in visitMul() 237 if (match(&I, m_Mul(m_Value(NewOp), m_Constant(C1)))) { in visitMul() 330 if (match(Op0, m_Neg(m_Value(X))) && match(Op1, m_Constant(Op1C))) in visitMul() 560 if (match(Op0, m_FNeg(m_Value(X))) && match(Op1, m_Constant(C))) in visitFMul() 571 if (match(Op1, m_Constant(C)) && C->isFiniteNonZeroFP()) { in visitFMul() 573 if (match(Op0, m_OneUse(m_FDiv(m_Constant(C1), m_Value(X))))) { in visitFMul() 580 if (match(Op0, m_FDiv(m_Value(X), m_Constant(C1)))) { in visitFMul() 598 if (match(Op0, m_OneUse(m_FAdd(m_Value(X), m_Constant(C1))))) { in visitFMul() 606 if (match(Op0, m_OneUse(m_FSub(m_Constant(C1), m_Value(X))))) { in visitFMul() [all …]
|
H A D | InstCombineShifts.cpp | 337 if (!match(I.getOperand(1), m_Constant(C1))) in foldShiftOfShiftedLogic() 350 m_OneUse(m_BinOp(ShiftOpcode, m_Value(X), m_Constant(C0)))) && in foldShiftOfShiftedLogic() 407 if (match(Op0, m_Constant(C)) && in commonShiftTransforms() 408 match(Op1, m_NUWAdd(m_Value(A), m_Constant(C1)))) { in commonShiftTransforms() 457 if (Op1->hasOneUse() && match(Op1, m_SRem(m_Value(A), m_Constant(C))) && in commonShiftTransforms() 729 if (match(Op0, m_BinOp(I.getOpcode(), m_Constant(C2), m_Value(X)))) in FoldShiftByConstant() 1116 if (match(Op1, m_Constant(C1))) { in visitShl() 1120 if (match(Op0, m_Mul(m_Value(X), m_Constant(C2)))) in visitShl() 1431 m_ZExtOrSelf(m_Sub(m_Constant(C1), in foldVariableSignZeroExtensionOfVariableHighBitExtract() 1433 m_ZExtOrSelf(m_Sub(m_Constant(C2), in foldVariableSignZeroExtensionOfVariableHighBitExtract() [all …]
|
H A D | InstCombineCasts.cpp | 693 if (match(BinOp0, m_Constant(C))) { in narrowBinOp() 699 if (match(BinOp1, m_Constant(C))) { in narrowBinOp() 723 if (match(BinOp0, m_Trunc(m_Value(A))) && match(BinOp1, m_Constant(C))) { in narrowBinOp() 877 if (match(Src, m_OneUse(m_LShr(m_Value(X), m_Constant(C))))) { in visitTrunc() 884 if (match(Src, m_OneUse(m_c_Or(m_LShr(m_Value(X), m_Constant(C)), in visitTrunc() 897 if (match(Src, m_LShr(m_SExt(m_Value(A)), m_Constant(C)))) { in visitTrunc() 942 if (match(Src, m_Shl(m_Value(A), m_Constant(C))) && in visitTrunc() 943 !match(A, m_Shr(m_Value(), m_Constant()))) { in visitTrunc() 1319 if (match(Src, m_OneUse(m_And(m_Trunc(m_Value(X)), m_Constant(C)))) && in visitZExt() 1325 if (match(Src, m_OneUse(m_Xor(m_Value(And), m_Constant(C)))) && in visitZExt() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
H A D | convert_tf_quant_ops_to_mhlo.cc | 94 if (!matchPattern(this_op.rhs(), m_Constant(&tensor_proto_attr))) in matchAndRewrite() 99 if (!matchPattern(this_op.rhs_scales(), m_Constant(&rhs_scales))) in matchAndRewrite() 104 if (!matchPattern(this_op.rhs_zps(), m_Constant(&rhs_zps))) in matchAndRewrite()
|
H A D | utils.h | 95 if (!matchPattern(value, m_Constant(&value_attr)) || in GetSplatValue() 104 if (!matchPattern(value, m_Constant(&value_attr)) || !value_attr.isSplat()) { in GetSplatValue()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
H A D | tfr_ops.cc | 393 if (matchPattern(cst_tensor_op.arg(), m_Constant(&array))) { in matchAndRewrite() 414 if (matchPattern(cst_tensor_op.arg(), m_Constant(&scalar))) { in matchAndRewrite() 519 if (!matchPattern(ge_op.index(), m_Constant(&index))) { in matchAndRewrite() 567 if (!matchPattern(operand, m_Constant(&array_elt))) { in matchAndRewrite() 744 if (!matchPattern(in_scale_op.arg(), m_Constant(&in_scale_attr)) || in matchAndRewrite() 754 if (!matchPattern(filter_scale_op.arg(), m_Constant(&filter_scale_attr))) { in matchAndRewrite() 805 if (!matchPattern(zp, m_Constant(&zp_attr))) { in matchAndRewrite()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | shape_inference.cc | 491 if (matchPattern(op, m_Constant())) return success(); in ComputeInputsRequiredForOutput() 525 if (matchPattern(op, m_Constant(&attr))) { in ComputeOutputComponent() 1429 matchPattern(op.window_dimensions(), m_Constant(&window_dimensions)) && in InferShapeForXlaReduceWindowOp() 1430 matchPattern(op.window_strides(), m_Constant(&window_strides)) && in InferShapeForXlaReduceWindowOp() 1431 matchPattern(op.base_dilations(), m_Constant(&base_dilations)) && in InferShapeForXlaReduceWindowOp() 1432 matchPattern(op.window_dilations(), m_Constant(&window_dilations)) && in InferShapeForXlaReduceWindowOp() 1433 matchPattern(op.padding(), m_Constant(&padding))) { in InferShapeForXlaReduceWindowOp() 1493 matchPattern(op.window_dimensions(), m_Constant(&window_dimensions)) && in InferShapeForXlaSelectAndScatterOp() 1494 matchPattern(op.window_strides(), m_Constant(&window_strides)) && in InferShapeForXlaSelectAndScatterOp() 1495 matchPattern(op.padding(), m_Constant(&padding))) { in InferShapeForXlaSelectAndScatterOp() [all …]
|
H A D | rewrite_util.h | 56 if (matchPattern(value, m_Constant(&float_attr)) && float_attr.isSplat() && in IsConstantValueOf() 61 if (matchPattern(value, m_Constant(&int_attr)) && int_attr.isSplat() && in IsConstantValueOf()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
H A D | tfl_ops.cc | 272 if (!matchPattern(value, m_Constant(&constant)) || !constant.isSplat()) { in EqualsZero() 1080 if (!matchPattern(input(), m_Constant(&input_tensor)) || in fold() 1081 !matchPattern(filter(), m_Constant(&weights_tensor)) || in fold() 1082 (has_bias && !matchPattern(bias(), m_Constant(&bias_tensor)))) { in fold() 1459 if (matchPattern(shape, m_Constant(&shape_value))) { in verify() 1684 if (!matchPattern(reshape.shape(), m_Constant(&shape))) { in matchAndRewrite() 1769 if (!matchPattern(shape, m_Constant(&shape_attr))) { in GetReshapeOutputType() 2027 if (matchPattern(op.begin(), m_Constant(&begin))) { in verify() 2039 if (matchPattern(op.size(), m_Constant(&size))) { in verify() 2076 if (!matchPattern(input_op, m_Constant(&attr))) { in NarrowDownInt64InputValuesForOp() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf_collective.cc | 110 if (!matchPattern(group_assignment_value, m_Constant(&group_assignment))) { in ConvertReplicaGroups() 295 if (!matchPattern(all_reduce.group_size(), m_Constant(&group_size_attr))) { in matchAndRewrite() 318 if (!matchPattern(all_reduce.group_key(), m_Constant(&group_key_attr))) { in matchAndRewrite()
|
H A D | legalize_tf.cc | 157 if (!matchPattern(value, m_Constant(&attrs)) || in GetIntegerHLOAxisFromTFAxis() 879 matchPattern(start_indices, m_Constant(&constant_start_indices)); in CanBeTranslatedToDynamicSlice() 899 if (!matchPattern(start_indices, m_Constant(&constant_start_indices))) { in TFSliceSizes2HLOSliceSizes() 1836 if (!matchPattern(op.k(), m_Constant(&kattr))) { in ExtractK() 2133 if (!matchPattern(op.fft_length(), m_Constant(&fft_length_attr))) { in matchAndRewrite() 2787 m_Constant(&orig_input_shape_attr))) { in matchAndRewrite() 3096 if (matchPattern(op.size(), m_Constant(&size_attr))) { in matchAndRewrite() 3322 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite() 3385 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite() 3501 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite() [all …]
|
/aosp_15_r20/external/llvm/lib/Transforms/InstCombine/ |
H A D | InstCombineMulDivRem.cpp | 201 if (match(&I, m_Mul(m_Shl(m_Value(NewOp), m_Constant(C2)), in visitMul() 202 m_Constant(C1))) && in visitMul() 216 if (match(&I, m_Mul(m_Value(NewOp), m_Constant(C1)))) { in visitMul() 283 if (match(Op0, m_OneUse(m_Add(m_Value(X), m_Constant(C1))))) { in visitMul() 1272 if (match(Op0, m_FMul(m_Value(X), m_Constant(C1)))) { in visitFDiv() 1278 } else if (match(Op0, m_FDiv(m_Value(X), m_Constant(C1)))) { in visitFDiv() 1311 if (match(Op1, m_FMul(m_Value(X), m_Constant(C2)))) in visitFDiv() 1313 else if (match(Op1, m_FDiv(m_Value(X), m_Constant(C2)))) { in visitFDiv() 1316 } else if (match(Op1, m_FDiv(m_Constant(C2), m_Value(X)))) { in visitFDiv()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
H A D | fake_quant_utils.h | 51 if (!matchPattern(min, m_Constant(&min_value))) { in operator() 54 if (!matchPattern(max, m_Constant(&max_value))) { in operator()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
H A D | fake_quant_utils.h | 51 if (!matchPattern(min, m_Constant(&min_value))) { in operator() 54 if (!matchPattern(max, m_Constant(&max_value))) { in operator()
|