Searched refs:tensor_has_rank_greater_or_equal_to (Results 1 – 4 of 4) sorted by relevance
| /external/executorch/kernels/portable/cpu/util/ |
| D | distance_util.cpp | 39 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(x1, 2)); in check_cdist_args()
40 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(x2, 2)); in check_cdist_args()
|
| D | copy_ops_util.cpp | 322 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 3)); in check_pixel_shuffle_args()
323 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(out, 3)); in check_pixel_shuffle_args()
335 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 3)); in check_pixel_unshuffle_args()
336 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(out, 3)); in check_pixel_unshuffle_args()
393 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 1)); in check_select_copy_out_args()
421 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 1)); in check_split_with_sizes_copy_args()
606 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 2)); in check_tril_args()
909 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 2)); in check_diagonal_copy_args()
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_roll.cpp | 22 ET_LOG_AND_RETURN_IF_FALSE(tensor_has_rank_greater_or_equal_to(in, 1)); in check_roll_args()
|
| /external/executorch/runtime/core/exec_aten/util/ |
| D | tensor_util.h | 604 inline bool tensor_has_rank_greater_or_equal_to( in tensor_has_rank_greater_or_equal_to() function
1356 using ::executorch::runtime::tensor_has_rank_greater_or_equal_to;
|