• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7    http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is distributed on an "AS IS" BASIS,
11WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16// Utility predicates that are shared by multiple passes.
17
18include "mlir/IR/OpBase.td"
19include "mlir/Dialect/StandardOps/IR/Ops.td"
20
21// Checks value is not produced by a TFL_Quant or
22// from TFL_Quant Op with same quant type.
23def NotFromQuantOpOrSameQuantType : Constraint<
24  CPred<"tflite::NotFromQuantOpOrSameQuantType($0,$1)">>;
25
26// Accepts two inputs and check if both have the same element type.
27def SameElementType : Constraint<
28  CPred<"getElementTypeOrSelf($0) == getElementTypeOrSelf($1)">>;
29
30// Checks if the value has only one user.
31def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>;
32