Home
last modified time | relevance | path

Searched refs:IsDimensionsDegenerateExceptLastOne (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dvalidators.cc108 bool IsDimensionsDegenerateExceptLastOne(ArrayRef<int64_t> elements_shape) { in IsDimensionsDegenerateExceptLastOne() function
117 bool IsDimensionsDegenerateExceptLastOne(Attribute val) { in IsDimensionsDegenerateExceptLastOne() function
119 return IsDimensionsDegenerateExceptLastOne(ranked_type.getShape()); in IsDimensionsDegenerateExceptLastOne()
Dvalidators.h108 bool IsDimensionsDegenerateExceptLastOne(mlir::Attribute val);
110 bool IsDimensionsDegenerateExceptLastOne(ArrayRef<int64_t> elements_shape);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Doptimize.td41 …t<CPred<"TFL::IsBroadcastableElementsAttrs($0, $1) && TFL::IsDimensionsDegenerateExceptLastOne($1)…
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc467 if (!IsDimensionsDegenerateExceptLastOne(shape)) return failure(); in matchAndRewrite()