Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc491 IntegerAttr axis_attr = rewriter.getI64IntegerAttr(axis); in matchAndRewrite() local
1500 IntegerAttr axis_attr = tf_pack_op.axisAttr(); in matchAndRewrite() local
1519 IntegerAttr axis_attr; in matchAndRewrite() local
2060 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
Dlegalize_tfl.cc1609 IntegerAttr axis_attr; in matchAndRewrite() local
2110 IntegerAttr axis_attr; in matchAndRewrite() local
2132 IntegerAttr axis_attr; in matchAndRewrite() local
2962 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
Dlegalize_common.cc2700 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1146 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
1377 DenseIntElementsAttr axis_attr; in Verify() local
2828 DenseIntElementsAttr axis_attr; in verify() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1561 DenseIntElementsAttr axis_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc2783 ElementsAttr axis_attr = axis_op->getAttrOfType<ElementsAttr>("value"); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc4062 auto axis_attr = axis->mutable_attr(); in RewriteGraph() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6283 DenseIntElementsAttr axis_attr; in matchAndRewrite() local