Searched refs:odsLoc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1079 auto odsLoc = rewriter.getFusedLoc({fused_batch_norm->getLoc()}); in matchAndRewrite() local
1146 odsLoc, ::mlir::DenseIntElementsAttr::get(reduce_dim_type, in matchAndRewrite()
1159 rewriter.create<::mlir::TF::ConstOp>(odsLoc, rest_size_inv_attr); in matchAndRewrite()
1166 odsLoc, x_value, reduce_dim_op, in matchAndRewrite()
1175 mul_op_1 = rewriter.create<::mlir::TF::MulOp>(odsLoc, tblgen_value_0, in matchAndRewrite()
1186 odsLoc, tblgen_value_0, tblgen_value_1); in matchAndRewrite()
1193 odsLoc, input_value, reduce_dim_op, in matchAndRewrite()
1202 mul_op_2 = rewriter.create<::mlir::TF::MulOp>(odsLoc, tblgen_value_0, in matchAndRewrite()
1214 rewriter.create<::mlir::TF::ConstOp>(odsLoc, in matchAndRewrite()
1222 add_op_1 = rewriter.create<::mlir::TF::AddOp>(odsLoc, in matchAndRewrite()
[all …]
|