Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc928 auto tf_conv_op = cast<TF::Conv2DBackpropInputOp>(op); in matchAndRewrite() local
931 tf_conv_op.out_backprop().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
933 tf_conv_op.filter().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
935 tf_conv_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
954 tf_conv_op.filter(), a1_filter_transpose_perm); in matchAndRewrite()
961 auto tmpAttr = tf_conv_op.strides(); in matchAndRewrite()
972 auto tmpAttr = tf_conv_op.dilations(); in matchAndRewrite()
984 if (!GetPaddingFromString(tf_conv_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
988 if (!FormatFromString(tf_conv_op.data_format().str(), &data_format_tf)) in matchAndRewrite()
993 tf_conv_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()
[all …]