Lines Matching defs:rank
177 const int64 rank = ShapeUtil::Rank(base_shape);
178 DimensionVector window_index(rank);
181 std::vector<int64> base_index(rank);
183 for (int64 i = 0; i < rank; ++i) {
295 ShapeUtil::Rank(broadcast->operand(0)->shape()), 0);
298 ShapeUtil::Rank(operand_to_broadcast.shape()))
300 << " and rank of operand_to_broadcast is: "
301 << ShapeUtil::Rank(operand_to_broadcast.shape());
909 const auto lhs_rank = ShapeUtil::Rank(lhs_shape);
910 const auto rhs_rank = ShapeUtil::Rank(rhs_shape);
1035 const auto lhs_rank = ShapeUtil::Rank(lhs->shape());
1036 const auto rhs_rank = ShapeUtil::Rank(rhs->shape());
1163 CHECK_EQ(ShapeUtil::Rank(pad->operand(0)->shape()),
1188 std::vector<int64> input_index(ShapeUtil::Rank(evaluated_operand.shape()),
1190 std::vector<int64> target_index(ShapeUtil::Rank(result->shape()), 0);
1440 TF_RET_CHECK(ShapeUtil::Rank(reduce->shape()) ==
1441 ShapeUtil::Rank(arg->shape()) - dimensions.size());
1471 const int64 result_rank = ShapeUtil::Rank(result->shape());
1555 int64 rank = ShapeUtil::Rank(operand_literal.shape());
1558 DimensionVector source_index(rank);
1666 DimensionVector operand_index(ShapeUtil::Rank(operand_literal.shape()));
1719 const int64 rank = ShapeUtil::Rank(operand->shape());
1722 DimensionVector operand_index(rank);
1723 for (int64 i = 0; i < rank; ++i) {
1928 std::vector<int64> result_index(ShapeUtil::Rank(result->shape()), 0);
2238 const int64 rank = ShapeUtil::Rank(reference_shape);
2241 CHECK_LT(concat_dim, rank);
2257 DimensionVector source_indices(rank, 0);
