Searched refs:rescale_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | convert_tfl_uint8.cc | 192 auto rescale_op = builder.create<tosa::RescaleOp>( in convert_graph_uint8_tensor() local
200 Operation *op_rescale_op = static_cast<Operation *>(rescale_op); in convert_graph_uint8_tensor()
202 tmp_val.replaceAllUsesWith(rescale_op.getResult()); in convert_graph_uint8_tensor()
315 auto rescale_op = builder.create<tosa::RescaleOp>( in convert_graph_uint8_tensor() local
323 Operation *op_rescale_op = static_cast<Operation *>(rescale_op); in convert_graph_uint8_tensor()
326 tmp_val.replaceAllUsesWith(rescale_op.getResult()); in convert_graph_uint8_tensor()
|
| D | legalize_utils.cc | 38 auto rescale_op = rewriter.create<tosa::RescaleOp>( in buildRescale() local
46 return rescale_op.getResult(); in buildRescale()
107 auto rescale_op = rewriter.create<tosa::RescaleOp>( in buildRescaleOpConvOutput() local
114 return rescale_op.getResult(); in buildRescaleOpConvOutput()
148 auto rescale_op = rewriter.create<tosa::RescaleOp>( in buildRescaleOpConvOutput() local
155 return rescale_op.getResult(); in buildRescaleOpConvOutput()
|
| D | legalize_common.cc | 669 Value rescale_op = buildRescale(rewriter, op, rescale_type, lhs_val, in convertConcatV2Op() local
673 lhs_val = rescale_op; in convertConcatV2Op()
679 Value rescale_op = buildRescale(rewriter, op, rescale_type, rhs_val, in convertConcatV2Op() local
683 rhs_val = rescale_op; in convertConcatV2Op()
714 Value rescale_op = buildRescale(rewriter, op, rescale_type, rhs_val, in convertConcatV2Op() local
718 rhs_val = rescale_op; in convertConcatV2Op()
|
| D | legalize_tfl.cc | 2815 Value rescale_op = buildRescale( in matchAndRewrite() local
2819 rewriter.replaceOp(op, {rescale_op}); in matchAndRewrite()
|