Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc40 auto larger_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in IsLegalNumpyRankedBroadcast() local
46 llvm::seq<int64_t>(larger_rank - smaller_rank, larger_rank); in IsLegalNumpyRankedBroadcast()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py117 larger_rank = array_ops.expand_dims(x[1], 0)
118 return larger_dim, larger_rank