Searched refs:in_rank (Results 1 – 8 of 8) sorted by relevance
82 int in_rank = in_shape.size(); in InitOffset() local87 for (int i = idx_lastshape; i < in_rank; ++i) { in InitOffset()90 std::vector<int> in_stride(in_rank); in InitOffset()91 in_stride[in_rank - 1] = 1; in InitOffset()92 for (int i = in_rank - 2; i >= 0; --i) { in InitOffset()
51 int in_rank = in_shape.size(); in DoGather() local53 MS_CHECK_LT(axis_, in_rank, RET_ERROR); in DoGather()60 for (int i = axis_ + 1; i < in_rank; ++i) { in DoGather()
90 int in_rank = static_cast<size_t>(in_shape.size()); in InitOffset() local99 for (int i = idx_lastshape; i < in_rank; ++i) { in InitOffset()102 std::vector<int> in_stride(in_rank); in InitOffset()103 in_stride[in_rank - 1] = 1; in InitOffset()104 for (int i = in_rank - 2; i >= 0; --i) { in InitOffset()
65 int in_rank = in_shape.size(); in DoGather() local67 MS_CHECK_LT(axis_, in_rank, RET_ERROR); in DoGather()82 for (int i = axis_ + 1; i < in_rank; ++i) { in DoGather()
38 int in_rank = (int)(input->shape_size_); in GatherNdInferShape() local40 if (indices->shape_[indices_rank - 1] > in_rank) { in GatherNdInferShape()49 for (i = indices->shape_[indices_rank - 1]; i < in_rank; ++i) { in GatherNdInferShape()
36 int in_rank = (int)(input->shape_size_); in UnsqueezeInferShape() local50 int sz = in_rank + dim_rank; in UnsqueezeInferShape()
46 int in_rank = static_cast<int>(in_shape.size()); in DoCode() local57 for (int i = axis + 1; i < in_rank; ++i) { in DoCode()
73 int in_rank = in_shape.size(); in DoGather() local82 for (int i = axis + 1; i < in_rank; ++i) { in DoGather()