// CHECK:  return %[[RES0]] : tensor<8xf32>
}

func.func @bitcast(%arg0: tensor<8xi32>) -> tensor<8xui32> {
  %0 = "tf.Bitcast"(%arg0) : (tensor<8xi32>) -> tensor<8xui32>
  func.return %0 : tensor<8xui32>

// CHECK-LABEL: bitcast
// CHECK: %[[RES0:.*]] = "tfl.bitcast"(%arg0) : (tensor<8xi32>) -> tensor<8xui32>
// CHECK:  return %[[RES0]] : tensor<8xui32>
}

func.func @testDoubleBitcast(%arg0: tensor<8x16x32x64xf32>) -> tensor<8x16x32x64xi32> {
  %0 = "tf.Bitcast"(%arg0) : (tensor<8x16x32x64xf32>) -> tensor<8x16x32x64x2xi16>
  %1 = "tf.Bitcast"(%0) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<8x16x32x64x2xi16>) -> tensor<8x16x32x64xi32>
  func.return %1: tensor<8x16x32x64xi32>

func.func @testBitcast(%arg0: tensor<3x4xui16>) -> tensor<3x4x!tf_type.quint16> {
  %0 = "tf.Bitcast"(%arg0) : (tensor<3x4xui16>) -> tensor<3x4x!tf_type.quint16>
  func.return %0 : tensor<3x4x!tf_type.quint16>
}

// -----

// CHECK-LABEL: func @testBitcast_v2
func.func @testBitcast_v2(%arg0: tensor<3x2xi16>) -> tensor<3xi32> {
  %0 = "tf.Bitcast"(%arg0) : (tensor<3x2xi16>) -> tensor<3xi32>
  func.return %0 : tensor<3xi32>
}

// -----

// CHECK: %[[BITCAST1:.+]] = "tfl.bitcast"(%[[ARG1]]) : (tensor<i64>) -> tensor<1xi64>
// CHECK: %[[MIN1:.+]] = "tfl.minimum"(%[[BITCAST1]], %[[MAX1]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>
// CHECK: %[[BITCAST2:.+]] = "tfl.bitcast"(%[[ARG2]]) : (tensor<i64>) -> tensor<1xi64>
// CHECK: %[[MIN2:.+]] = "tfl.minimum"(%[[BITCAST2]], %[[MAX2]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>

  let hasFolder = 1;
}

def TFL_BitcastOp : TFL_Op<"bitcast", [Pure]> {
  let summary = "Bitcast operator";

  let description = [{
    Bitcasts a tensor from one type to another.
  }];

  let arguments = (ins AnyTensor:$input);

  let results = (outs AnyTensor:$output);

  // TFLite's bitcast bitop does not utilize options, instead derives types
  // from the TfLiteTensors.

  // CHECK: return %0 : tensor<8xf32>
}


// -----

// CHECK-LABEL: testBitcast
func.func @testBitcast(%arg0: tensor<8xui32>) -> tensor<8xi32> {
  // CHECK: "tfl.bitcast"(%arg0)
  %0 = "tfl.bitcast"(%arg0) : (tensor<8xui32>) -> tensor<8xi32>
  func.return %0 : tensor<8xi32>
  // CHECK: return %0 : tensor<8xi32>
}

// -----

// CHECK-LABEL: testBitwiseXor