You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
4.2 KiB
78 lines
4.2 KiB
/**
|
|
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef INC_GRAPH_UTILS_TENSOR_UTILS_H_
|
|
#define INC_GRAPH_UTILS_TENSOR_UTILS_H_
|
|
|
|
#include <vector>
|
|
#include "graph/def_types.h"
|
|
#include "graph/ge_error_codes.h"
|
|
#include "graph/ge_tensor.h"
|
|
|
|
namespace ge {
|
|
class TensorUtils {
|
|
public:
|
|
static ge::graphStatus GetSize(const GeTensorDesc &tensorDesc, int64_t &size);
|
|
static void SetSize(GeTensorDesc &tensorDesc, int64_t size);
|
|
static uint32_t GetWeightSize(const ConstGeTensorPtr &tensorPtr);
|
|
static uint32_t GetWeightSize(const GeTensor &tensor);
|
|
static uint32_t GetWeightSize(const GeTensorDesc &tensorDesc);
|
|
static uint8_t *GetWeightAddr(const ConstGeTensorPtr &tensorPtr, uint8_t *base);
|
|
static uint8_t *GetWeightAddr(const GeTensor &tensor, uint8_t *base);
|
|
static void SetWeightSize(GeTensorDesc &tensorDesc, uint32_t size);
|
|
static ge::graphStatus GetReuseInput(const GeTensorDesc &tensorDesc, bool &flag);
|
|
static void SetReuseInput(GeTensorDesc &tensorDesc, bool flag);
|
|
static ge::graphStatus GetOutputTensor(const GeTensorDesc &tensorDesc, bool &flag);
|
|
static void SetOutputTensor(GeTensorDesc &tensorDesc, bool flag);
|
|
static graphStatus GetDeviceType(const GeTensorDesc &tensorDesc, DeviceType &type);
|
|
static void SetDeviceType(GeTensorDesc &tensorDesc, DeviceType type);
|
|
static ge::graphStatus GetInputTensor(const GeTensorDesc &tensorDesc, bool &flag);
|
|
static void SetInputTensor(GeTensorDesc &tensorDesc, bool flag);
|
|
static ge::graphStatus GetRealDimCnt(const GeTensorDesc &tensorDesc, uint32_t &cnt);
|
|
static void SetRealDimCnt(GeTensorDesc &tensorDesc, uint32_t cnt);
|
|
static ge::graphStatus GetReuseInputIndex(const GeTensorDesc &tensorDesc, uint32_t &idx);
|
|
static void SetReuseInputIndex(GeTensorDesc &tensorDesc, uint32_t idx);
|
|
static ge::graphStatus GetDataOffset(const GeTensorDesc &tensorDesc, int64_t &offset);
|
|
static void SetDataOffset(GeTensorDesc &tensorDesc, int64_t offset);
|
|
static ge::graphStatus GetCmpsSize(const GeTensorDesc &tensorDesc, uint32_t &cmp_size);
|
|
static void SetCmpsSize(GeTensorDesc &tensorDesc, uint32_t cmp_size);
|
|
static ge::graphStatus GetCmpsTab(const GeTensorDesc &tensorDesc, vector<uint8_t> &vec);
|
|
static void SetCmpsTab(GeTensorDesc &tensorDesc, const uint8_t *data, size_t size);
|
|
static ge::graphStatus GetCmpsTabOffset(const GeTensorDesc &tensorDesc, int64_t &tab_offset);
|
|
static void SetCmpsTabOffset(GeTensorDesc &tensorDesc, int64_t tab_offset);
|
|
static ge::graphStatus GetCmpsInfo(const GeTensorDesc &tensorDesc, CompressInfo &info);
|
|
static void SetCmpsInfo(GeTensorDesc &tensorDesc, const CompressInfo &info);
|
|
static bool HasAlloffsetQuantizeInfo(const GeTensorDesc &tensorDesc);
|
|
static ge::graphStatus GetAlloffsetQuantizeInfo(const GeTensorDesc &tensorDesc, AllOffsetQuantizeInfo &info);
|
|
static void SetAlloffsetQuantizeInfo(GeTensorDesc &tensorDesc, const AllOffsetQuantizeInfo &info);
|
|
static ge::graphStatus GetRC(const GeTensorDesc &tensorDesc, uint32_t &rc);
|
|
static void SetRC(GeTensorDesc &tensorDesc, uint32_t rc);
|
|
|
|
///
|
|
/// calculate tensor mem size.
|
|
/// @param shape tensor shape
|
|
/// @param format tensor format
|
|
/// @param data_type tensor data type
|
|
/// @param mem_size -1 means unknown shape,other means mem size
|
|
/// @return GRAPH_SUCCESS:success, other:failed
|
|
///
|
|
static ge::graphStatus CalcTensorMemSize(const GeShape &shape, Format format, DataType data_type, int64_t &mem_size);
|
|
static ge::graphStatus GetTensorMemorySizeInBytes(const GeTensorDesc &desc_temp, int64_t &size_temp);
|
|
static ge::graphStatus GetTensorSizeInBytes(const GeTensorDesc &desc_temp, int64_t &size_temp);
|
|
};
|
|
} // namespace ge
|
|
#endif // INC_GRAPH_UTILS_TENSOR_UTILS_H_
|