Home
last modified time | relevance | path

Searched defs:axis_attr (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc490 IntegerAttr axis_attr = rewriter.getI64IntegerAttr(axis); in matchAndRewrite() local
1507 IntegerAttr axis_attr = tf_pack_op.axisAttr(); in matchAndRewrite() local
1526 IntegerAttr axis_attr; in matchAndRewrite() local
2065 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
Dlegalize_tfl.cc1406 IntegerAttr axis_attr; in matchAndRewrite() local
1909 IntegerAttr axis_attr; in matchAndRewrite() local
1932 IntegerAttr axis_attr; in matchAndRewrite() local
2616 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
Dlegalize_common.cc2441 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1087 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
1308 DenseIntElementsAttr axis_attr; in Verify() local
2657 DenseIntElementsAttr axis_attr; in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1569 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1157 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
6515 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc4062 auto axis_attr = axis->mutable_attr(); in RewriteGraph() local