Searched refs:IsTailOfShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | move_binary_operator_before_reshape.cc | 25 bool IsTailOfShape(const Shape& tail, const Shape& shape) { in IsTailOfShape() function
106 if (!IsTailOfShape( in Run()
135 if (!IsTailOfShape( in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_patterns.td | 367 def IsTailOfShape : Constraint<CPred<
368 "TFL::IsTailOfShape($0.getType(), $1.getType())">>;
416 // couldn't be reshaped to [1,40,40]. `IsTailOfShape` constraint is added to
426 [(IsTailOfShape $rhs, $lhs),
431 (IsTailOfShape $rhs, $input),
448 // couldn't be reshaped to [1,40,40]. `IsTailOfShape` constraint is added to
458 [(IsTailOfShape $rhs, $lhs),
463 (IsTailOfShape $rhs, $input),
704 (IsTailOfShape $input1, $constant),
705 (IsTailOfShape $constant, $input1),
[all …]
|
| D | optimize.cc | 109 bool IsTailOfShape(Type type1, Type type2) { in IsTailOfShape() function
|