| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 174 Value input_value, ElementsAttr axes_elems, bool keep_dims);
179 Value input_value, ElementsAttr axes_elems, bool keep_dims);
184 Value input_value, ElementsAttr axes_elems, bool keep_dims);
189 Value input_value, ElementsAttr axes_elems, bool keep_dims);
194 Value input_value, ElementsAttr axes_elems, bool keep_dims);
199 Value input_value, ElementsAttr axes_elems, bool keep_dims);
204 Value input_value, ElementsAttr axes_elems, bool keep_dims);
|
| D | legalize_tf.cc | 1038 ElementsAttr axes_elems; in matchAndRewrite() local
1039 if (!matchPattern(tf_all_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite()
1047 rewriter, op, output_type, tf_all_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1064 ElementsAttr axes_elems; in matchAndRewrite() local
1065 if (!matchPattern(tf_any_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite()
1073 rewriter, op, output_type, tf_any_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1090 ElementsAttr axes_elems; in matchAndRewrite() local
1091 if (!matchPattern(tf_max_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite()
1099 rewriter, op, output_type, tf_max_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1116 ElementsAttr axes_elems; in matchAndRewrite() local
[all …]
|
| D | legalize_common.cc | 2311 Value input_value, ElementsAttr axes_elems, bool keep_dims, in convertReduceOpCommon() argument
2323 if (axes_elems.getNumElements() == 0) { in convertReduceOpCommon()
2336 for (int i = 0; i < axes_elems.getNumElements(); i++) { in convertReduceOpCommon()
2337 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceOpCommon()
2373 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAllOp() argument
2379 rewriter, op, output_type, input_value, axes_elems, keep_dims, in convertReduceAllOp()
2386 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAnyOp() argument
2392 rewriter, op, output_type, input_value, axes_elems, keep_dims, in convertReduceAnyOp()
2399 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMinOp() argument
2405 rewriter, op, output_type, input_value, axes_elems, keep_dims, in convertReduceMinOp()
[all …]
|
| D | legalize_tfl.cc | 1740 ElementsAttr axes_elems; in matchAndRewrite() local
1741 if (!matchPattern(tfl_any_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite()
1749 rewriter, op, output_type, tfl_any_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1766 ElementsAttr axes_elems; in matchAndRewrite() local
1767 if (!matchPattern(tfl_max_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite()
1775 rewriter, op, output_type, tfl_max_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1792 ElementsAttr axes_elems; in matchAndRewrite() local
1793 if (!matchPattern(tfl_min_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite()
1801 rewriter, op, output_type, tfl_min_op.input(), axes_elems, keep_dims); in matchAndRewrite()
1818 ElementsAttr axes_elems; in matchAndRewrite() local
[all …]
|