Home
last modified time | relevance | path

Searched refs:tf_maxpool_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc565 auto tf_maxpool_op = cast<TF::MaxPoolOp>(op); in matchAndRewrite() local
568 tf_maxpool_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
570 tf_maxpool_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
574 auto tmpAttr = tf_maxpool_op.data_formatAttr(); in matchAndRewrite()
581 auto tmpAttr = tf_maxpool_op.strides(); in matchAndRewrite()
592 auto tmpAttr = tf_maxpool_op.ksize(); in matchAndRewrite()
604 if (!GetPaddingFromString(tf_maxpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
612 for (auto& elem : tf_maxpool_op.ksize()) { in matchAndRewrite()
629 op, output_type, tf_maxpool_op.input(), kernel, stride, pad); in matchAndRewrite()