Searched refs:rescale_op (Results 1 – 5 of 5) sorted by relevance
191 auto rescale_op = builder.create<tosa::RescaleOp>( in convert_graph_uint8_tensor() local199 Operation *op_rescale_op = static_cast<Operation *>(rescale_op); in convert_graph_uint8_tensor()201 tmp_val.replaceAllUsesWith(rescale_op.getResult()); in convert_graph_uint8_tensor()309 auto rescale_op = builder.create<tosa::RescaleOp>( in convert_graph_uint8_tensor() local317 Operation *op_rescale_op = static_cast<Operation *>(rescale_op); in convert_graph_uint8_tensor()320 tmp_val.replaceAllUsesWith(rescale_op.getResult()); in convert_graph_uint8_tensor()
43 auto rescale_op = CreateOpAndInfer<tosa::RescaleOp>( in buildRescale() local51 return rescale_op.getResult(); in buildRescale()114 auto rescale_op = CreateOpAndInfer<tosa::RescaleOp>( in buildRescaleOpConvOutput() local121 return rescale_op.getResult(); in buildRescaleOpConvOutput()150 auto rescale_op = CreateOpAndInfer<tosa::RescaleOp>( in buildRescaleOpConvOutput() local157 return rescale_op.getResult(); in buildRescaleOpConvOutput()
641 Value rescale_op = buildRescale( in convertConcatV2Op() local644 values_rescaled.push_back(rescale_op); in convertConcatV2Op()
3001 Value rescale_op = in matchAndRewrite() local3006 rewriter.replaceOp(op, {rescale_op}); in matchAndRewrite()
788 LogicalResult matchAndRewrite(TFRQuantRescaleOp rescale_op, in matchAndRewrite() argument790 Value input = rescale_op.input(); in matchAndRewrite()791 Value scale = rescale_op.scale(); in matchAndRewrite()792 Value zp = rescale_op.zp(); in matchAndRewrite()794 const Location loc = rescale_op->getLoc(); in matchAndRewrite()795 const auto result_types = rescale_op->getResultTypes(); in matchAndRewrite()814 rewriter.setInsertionPoint(rescale_op); in matchAndRewrite()838 rescale_op.output().replaceAllUsesWith(cast_output_to_i32.getResult(0)); in matchAndRewrite()