Searched refs:HasRankAtMost (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_patterns.td | 43 class HasRankAtMost<int n> : Constraint<
419 (HasRankAtMost<4> $input),
420 (HasRankAtMost<4> $operand)]>;
427 (HasRankAtMost<4> $operand),
428 (HasRankAtMost<4> $input)]>;
473 (HasRankAtMost<4> $input),
474 (HasRankAtMost<4> $lhs),
475 (HasRankAtMost<4> $rhs),
511 (HasRankAtMost<4> $input),
512 (HasRankAtMost<4> $lhs),
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_tensor_helper.h | 58 inline bool HasRankAtMost(Value value, int64_t rank) { in HasRankAtMost() function
|
| D | tf_ops_n_z.cc | 547 if (!HasRankAtMost(op.input(), std::numeric_limits<int32_t>::max())) in verify()
1201 if (!HasRankAtMost(op.input(), std::numeric_limits<int32_t>::max())) in verify()
2908 if (!HasRankAtMost(op.num_segments(), 0)) in VerifyUnsortedSegmentReduction()
|
| D | tf_ops_a_m.cc | 1008 if (!HasRankAtMost(axis, 1)) { in Verify()
2825 if (!HasRankAtMost(op.axis(), 1)) in verify()
|