Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc2405 llvm::Optional<llvm::ArrayRef<int64_t>> shape_constraint = shape(); in fold() local
2406 if (type.getShape() == shape_constraint) return input(); in fold()
2410 if (shape_constraint.has_value() && in fold()
2411 shape_constraint->size() == type.getShape().size()) { in fold()
2412 for (int i = 0; i < shape_constraint->size(); ++i) { in fold()
2413 if (!ShapedType::isDynamic(shape_constraint.getValue()[i]) && in fold()
2414 type.getDimSize(i) != shape_constraint.getValue()[i]) { in fold()