[Functions] Add quantile function (#700)

* Add sort function

* Add isfinite function

* upgrade isinf isnan

* Add Scalar to FDTensor

* Add floor, ceil function

* add cast functions

* Update out_tmp

* Update quantile

* add gather scatter along axis

* finish quantile function

* Add quantile unittest

* refresh code style for test source code

* Add comments

* Add full function

* Add scalar to fd tensor

* Add full unittest

* Add functions headers

* move fdtensor operators to fastdeploy namespace
This commit is contained in:
Jack Zhou
2022-11-28 09:51:40 +08:00
committed by GitHub
parent 4e74ac06fb
commit 129dda7809
37 changed files with 1567 additions and 75 deletions

View File

@@ -0,0 +1,42 @@
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "fastdeploy/function/full.h"
#include "fastdeploy/function/eigen.h"
#include <algorithm>
namespace fastdeploy {
namespace function {
template <typename T> void FullValue(FDTensor* tensor, const Scalar& val) {
auto t = EigenVector<T>::Flatten(*tensor);
auto& place = *EigenDeviceWrapper::GetInstance()->GetDevice();
t.device(place) = t.constant(val.to<T>());
}
void Full(const Scalar& value, const std::vector<int64_t>& shape, FDTensor* out,
FDDataType dtype) {
FD_VISIT_ALL_TYPES(dtype, "Full", ([&] {
out->Allocate(shape, dtype);
FullValue<data_t>(out, value);
}));
}
void FullLike(const FDTensor& x, const Scalar& value, FDTensor* out,
FDDataType dtype) {
Full(value, x.Shape(), out, dtype);
}
} // namespace function
} // namespace fastdeploy