Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc29 Value buildRescale(PatternRewriter& rewriter, Operation* op, in buildRescale() function
59 return buildRescale(rewriter, op, output_type, input_val, input_scale, in buildRescaleToInt32()
74 return buildRescale(rewriter, op, output_type, input_val, output_scale, 0, in buildRescaleFromInt32()
Dlegalize_utils.h38 Value buildRescale(PatternRewriter& rewriter, Operation* op,
Dlegalize_common.cc669 Value rescale_op = buildRescale(rewriter, op, rescale_type, lhs_val, in convertConcatV2Op()
679 Value rescale_op = buildRescale(rewriter, op, rescale_type, rhs_val, in convertConcatV2Op()
714 Value rescale_op = buildRescale(rewriter, op, rescale_type, rhs_val, in convertConcatV2Op()
1452 buildRescale(rewriter, op, int32_logits_type, logits_value, 1.0f, in convertSoftmaxOp()
1475 Value op4_rescale_op3 = buildRescale( in convertSoftmaxOp()
1546 Value op15_rescale_op14 = buildRescale(rewriter, op, int32_rsum_type, in convertSoftmaxOp()
1551 buildRescale(rewriter, op, int32_logits_type, op5_table_op4.getResult(), in convertSoftmaxOp()
1567 return buildRescale(rewriter, op, output_type, in convertSoftmaxOp()
2686 return buildRescale(rewriter, op, output_type, op3_cast_op2.getResult(), 1.0, in convertQuantizeOp()
2715 buildRescale(rewriter, op, output_int32_type, input_value, 1.0, 0, 0); in convertDequantizeOp()
Dlegalize_tfl.cc2485 buildRescale(rewriter, op, int16_type, tfl_hardswish_op.input(), in matchAndRewrite()
2495 buildRescale(rewriter, op, output_type, op2_table_op1.getResult(), in matchAndRewrite()
2499 Value op4_rescale_in = buildRescale(rewriter, op, int32_type, in matchAndRewrite()
2603 buildRescale(rewriter, op, int16_type, tfl_logistic_op.x(), in matchAndRewrite()
2614 buildRescale(rewriter, op, output_type, op2_table_op1.getResult(), in matchAndRewrite()
2675 buildRescale(rewriter, op, int16_type, tfl_tanh_op.input(), in matchAndRewrite()
2686 buildRescale(rewriter, op, output_type, op2_table_op1.getResult(), in matchAndRewrite()
2815 Value rescale_op = buildRescale( in matchAndRewrite()