Searched refs:new_cst (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 366 Operation *new_cst = nullptr; in matchAndRewrite() local
379 new_cst = rewriter.create<TF::ConstOp>(loc, new_out_type, attr); in matchAndRewrite()
381 new_cst = rewriter.create<CastOp>(loc, out_type, new_cst->getResult(0)); in matchAndRewrite()
383 rewriter.replaceOp(cst_tensor_op, new_cst->getResult(0)); in matchAndRewrite()
390 new_cst = rewriter.create<TF::ConstOp>(loc, new_out_type, scalar); in matchAndRewrite()
392 new_cst = rewriter.create<CastOp>(loc, out_type, new_cst->getResult(0)); in matchAndRewrite()
394 rewriter.replaceOp(cst_tensor_op, new_cst->getResult(0)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 473 auto new_cst = cst.reshape(RankedTensorType::get( in matchAndRewrite() local
475 Type new_type = new_cst.getType(); in matchAndRewrite()
481 rewriter.create<ConstantOp>(mul_op.getLoc(), new_type, new_cst); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_driver.cc | 718 auto new_cst = in PreprocessConstantOps() local
720 user->setOperand(operand_num, new_cst); in PreprocessConstantOps()
|