experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add quantization type and traits for result for resizeNearestNeighbor

Browse Source 
PiperOrigin-RevId: 263774102
This commit is contained in:
Karim Nosir 2019-08-16 08:18:41 -07:00 committed by TensorFlower Gardener
parent 54b01260fe
commit b237937548
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
View File

@ -2266,7 +2266,8 @@ def TFL_ResizeBilinearOp: TFL_Op<"resize_bilinear", [
}
def TFL_ResizeNearestNeighborOp : TFL_Op<"resize_nearest_neighbor",
[NoSideEffect]> {
[NoSideEffect,
TFL_SameOperandsAndResultsScale]> {
let summary = "ResizeNearestNeighbor Op";
let description = [{
@ -2274,13 +2275,13 @@ def TFL_ResizeNearestNeighborOp : TFL_Op<"resize_nearest_neighbor",
}];
let arguments = (ins
TensorOf<[F32, I8, TFL_Uint8]>:$input,
TensorOf<[F32, I8, TFL_Uint8, TFL_QUI8, TFL_QI8]>:$input,
TensorOf<[I32]>:$size,
BoolAttr:$align_corners
);
let results = (outs
TensorOf<[F32, I8, TFL_Uint8]>:$output
TensorOf<[F32, I8, TFL_Uint8, TFL_QUI8, TFL_QI8]>:$output
);
let hasOptions = 1;

6
tensorflow/compiler/mlir/lite/tests/ops.mlir
View File

@ -196,6 +196,12 @@ func @testQuantizedSquare(tensor<? x !quant.uniform<u8:f32, 0.1>>) -> tensor<? x
return %0 : tensor<? x !quant.uniform<u8:f32, 0.1>>
}
func @testQuantizedResizeNearestNeighbor(tensor<? x !quant.uniform<u8:f32, 0.1>>, tensor<? x i32>) -> tensor<? x !quant.uniform<u8:f32, 0.1>> {
^bb0(%arg0: tensor<? x !quant.uniform<u8:f32, 0.1>>, %arg1: tensor<? x i32>):
%0 = "tfl.resize_nearest_neighbor"(%arg0, %arg1) { align_corners = false } : (tensor<? x !quant.uniform<u8:f32, 0.1>>, tensor<? x i32>) -> tensor<? x !quant.uniform<u8:f32, 0.1>>
return %0 : tensor<? x !quant.uniform<u8:f32, 0.1>>
}
// CHECK-LABEL: testTanh
func @testTanh(tensor<? x f32>) -> tensor<? x f32> {
^bb0(%arg0: tensor<? x f32>):