Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ops/
Dsoftmax.cc42 CheckAndConvertUtils::CheckInRange<int64_t>("axis", item, kIncludeLeft, {-rank, rank}, op_name); in Init()
63 CheckAndConvertUtils::CheckInRange<int64_t>("axis", item, kIncludeLeft, {-rank, rank}, op_name); in SoftMaxInferShape()
Droll.cc37 …CheckAndConvertUtils::CheckInRange("axis value", axis, kIncludeLeft, {-x_rank, x_rank}, prim_name); in InferShape()
Dl2_normalize.cc53 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in L2NormalizeInfer()
Darg_min.cc48 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in ArgMinInfer()
Dlog_softmax.cc50 CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-rank, rank}, op_name); in LogSoftmaxInferShape()
Darg_max.cc28 …CheckAndConvertUtils::CheckInRange<int64_t>("argmax axis", axis, kIncludeLeft, {-x_rank, x_rank}, … in InferShape()
Dreduce.cc30 CheckAndConvertUtils::CheckInRange("axis", one_axis, kIncludeLeft, {-dim, dim}, "Reduce"); in reduce_one_axis()
Dgather.cc68 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis_val, kIncludeLeft, {-params_rank, params_… in GatherInfer()
Dsplit_v.cc35 …CheckAndConvertUtils::CheckInRange("split_dim", split_dim, kIncludeLeft, {-x_rank, x_rank}, prim_n… in InferShape()
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.h50 kIncludeLeft = 2, // [a,b) enumerator
138 …{kIncludeLeft, [](T num1, std::pair<T, T> range) -> bool { return num1 >= range.first && num1 < ra…
149 {kIncludeLeft, {"in [", ")"}},