Searched refs:HasRankAtLeast (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_tensor_helper.h | 51 inline bool HasRankAtLeast(Value value, int64_t rank) { in HasRankAtLeast() function
|
| D | tf_ops_a_m.cc | 213 if (!HasRankAtLeast(op.x(), 2)) { in Verify()
216 if (!HasRankAtLeast(op.y(), 2)) { in Verify()
506 if (!HasRankAtLeast(op.value(), 2)) in verify()
513 if (!HasRankAtLeast(op.value(), 3)) in verify()
568 if (!HasRankAtLeast(op.out_backprop(), 2)) in verify()
575 if (!HasRankAtLeast(op.out_backprop(), 3)) in verify()
2600 if (!HasRankAtLeast(inputs, 1)) in verify()
3064 if (!HasRankAtLeast(op.input(), 2)) { in verify()
|
| D | tf_ops_n_z.cc | 544 if (!HasRankAtLeast(op.input(), 2)) in verify()
1334 if (!HasRankAtLeast(op.logits(), 1)) { in verify()
2403 if (!HasRankAtLeast(op.tensor(), 1)) in verify()
2406 if (!HasRankAtLeast(op.indices(), 1)) in verify()
2512 if (!HasRankAtLeast(op.input(), 1)) in verify()
|