Searched refs:kIncludeLeft (Results 1 – 10 of 10) sorted by relevance
/third_party/mindspore/mindspore/core/ops/ |
D | softmax.cc | 42 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()
|
D | roll.cc | 37 …CheckAndConvertUtils::CheckInRange("axis value", axis, kIncludeLeft, {-x_rank, x_rank}, prim_name); in InferShape()
|
D | l2_normalize.cc | 53 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in L2NormalizeInfer()
|
D | arg_min.cc | 48 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in ArgMinInfer()
|
D | log_softmax.cc | 50 CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-rank, rank}, op_name); in LogSoftmaxInferShape()
|
D | arg_max.cc | 28 …CheckAndConvertUtils::CheckInRange<int64_t>("argmax axis", axis, kIncludeLeft, {-x_rank, x_rank}, … in InferShape()
|
D | reduce.cc | 30 CheckAndConvertUtils::CheckInRange("axis", one_axis, kIncludeLeft, {-dim, dim}, "Reduce"); in reduce_one_axis()
|
D | gather.cc | 68 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis_val, kIncludeLeft, {-params_rank, params_… in GatherInfer()
|
D | split_v.cc | 35 …CheckAndConvertUtils::CheckInRange("split_dim", split_dim, kIncludeLeft, {-x_rank, x_rank}, prim_n… in InferShape()
|
/third_party/mindspore/mindspore/core/utils/ |
D | check_convert_utils.h | 50 kIncludeLeft = 2, // [a,b) enumerator 138 …{kIncludeLeft, [](T num1, std::pair<T, T> range) -> bool { return num1 >= range.first && num1 < ra… 149 {kIncludeLeft, {"in [", ")"}},
|