Searched refs:buildRescale (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
D | legalize_utils.cc | 29 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()
|
D | legalize_utils.h | 38 Value buildRescale(PatternRewriter& rewriter, Operation* op,
|
D | legalize_common.cc | 669 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()
|
D | legalize_tfl.cc | 2485 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()
|