Searched refs:GetPositiveAxis (Results 1 – 4 of 4) sorted by relevance
50 int64_t GetPositiveAxis(int64_t axis_value, size_t increment);
218 int64_t GetPositiveAxis(int64_t axis_value, size_t increment) { in GetPositiveAxis() function
112 axis_value = GetPositiveAxis(axis_value, LongToSize(rank_base + 1)); in InferImplStack()845 uint64_t axis_value_pos = LongToUlong(GetPositiveAxis(axis_value, LongToSize(rank))); in InferImplSplit()962 axis_value = GetPositiveAxis(axis_value, LongToSize(rank_base)); in InferImplConcat()
68 begin_params_axis = abstract::GetPositiveAxis(begin_params_axis, input_rank); in LayerNormInfer()