Searched refs:tf_conv2d_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | fuse_bias_tf.cc | 76 TF::Conv2DOp tf_conv2d_op = in matchAndRewrite() local
79 if (!tf_conv2d_op) { in matchAndRewrite()
85 tf_conv2d_op.filter().getType().dyn_cast<RankedTensorType>().getShape(); in matchAndRewrite()
97 rewriter, op, output_type, tf_conv2d_op.input(), tf_conv2d_op.filter(), in matchAndRewrite()
98 bias, tf_conv2d_op.strides(), tf_conv2d_op.dilations(), in matchAndRewrite()
99 tf_conv2d_op.explicit_paddings(), tf_conv2d_op.padding(), in matchAndRewrite()
100 tf_conv2d_op.data_format()); in matchAndRewrite()
|
| D | legalize_tf.cc | 816 auto tf_conv2d_op = cast<TF::Conv2DOp>(op); in matchAndRewrite() local
819 tf_conv2d_op.filter().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
821 tf_conv2d_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
832 rewriter, op, output_type, tf_conv2d_op.input(), tf_conv2d_op.filter(), in matchAndRewrite()
833 bias, tf_conv2d_op.strides(), tf_conv2d_op.dilations(), in matchAndRewrite()
834 tf_conv2d_op.explicit_paddings(), tf_conv2d_op.padding(), in matchAndRewrite()
835 tf_conv2d_op.data_format()); in matchAndRewrite()
|