Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_clustering.cc168 if (auto result_constraint = results.GetConstraint(result)) { in MatchAndUpdateConstraints() local
173 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
176 ? Merge(*default_constraint, *result_constraint) in MatchAndUpdateConstraints()
177 : *result_constraint; in MatchAndUpdateConstraints()
202 if (auto result_constraint = results.GetConstraint(op.getResult())) { in MatchAndUpdateConstraints() local
203 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
205 if (*result_constraint == ValueConstraint::kShape) { in MatchAndUpdateConstraints()
350 if (auto result_constraint = results.GetConstraint(op->getResult(0))) { in MatchAndUpdateConstraints() local
351 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
352 operands.Insert(op->getOperand(0), *result_constraint); in MatchAndUpdateConstraints()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_clustering.cc173 if (auto result_constraint = results.GetConstraint(result)) { in MatchAndUpdateConstraints() local
178 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
181 ? Merge(*default_constraint, *result_constraint) in MatchAndUpdateConstraints()
182 : *result_constraint; in MatchAndUpdateConstraints()
214 if (auto result_constraint = results.GetConstraint(op.getResult())) { in MatchAndUpdateConstraints() local
215 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
217 if (*result_constraint == ValueConstraint::kShape) { in MatchAndUpdateConstraints()
363 if (auto result_constraint = results.GetConstraint(op->getResult(0))) { in MatchAndUpdateConstraints() local
364 if (*result_constraint == ValueConstraint::kValue) return failure(); in MatchAndUpdateConstraints()
365 operands.Insert(op->getOperand(0), *result_constraint); in MatchAndUpdateConstraints()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtest_cluster_ops_by_policy.cc44 if (auto result_constraint = results.GetConstraint(op->getResult(0))) { in MatchAndUpdateConstraints() local
46 if (is_op_a && *result_constraint == ValueConstraint::kShape) in MatchAndUpdateConstraints()
50 if (*result_constraint == ValueConstraint::kValue) in MatchAndUpdateConstraints()