Home
last modified time | relevance | path

Searched defs:axes_elems (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1038 ElementsAttr axes_elems; in matchAndRewrite() local
1064 ElementsAttr axes_elems; in matchAndRewrite() local
1090 ElementsAttr axes_elems; in matchAndRewrite() local
1116 ElementsAttr axes_elems; in matchAndRewrite() local
1142 ElementsAttr axes_elems; in matchAndRewrite() local
1168 ElementsAttr axes_elems; in matchAndRewrite() local
1194 ElementsAttr axes_elems; in matchAndRewrite() local
Dlegalize_common.cc2311 Value input_value, ElementsAttr axes_elems, bool keep_dims, in convertReduceOpCommon()
2373 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAllOp()
2386 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAnyOp()
2399 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMinOp()
2412 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMaxOp()
2425 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceProdOp()
2450 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceSumOp()
2500 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMeanOp()
Dlegalize_tfl.cc1740 ElementsAttr axes_elems; in matchAndRewrite() local
1766 ElementsAttr axes_elems; in matchAndRewrite() local
1792 ElementsAttr axes_elems; in matchAndRewrite() local
1818 ElementsAttr axes_elems; in matchAndRewrite() local
1844 ElementsAttr axes_elems; in matchAndRewrite() local
1870 ElementsAttr axes_elems; in matchAndRewrite() local