Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc39 auto smaller_rank = std::min(lhs_type.getRank(), rhs_type.getRank()); in IsLegalNumpyRankedBroadcast() local
42 if (smaller_rank != broadcast_dims.getNumElements()) { in IsLegalNumpyRankedBroadcast()
46 llvm::seq<int64_t>(larger_rank - smaller_rank, larger_rank); in IsLegalNumpyRankedBroadcast()