graphengine/inc/graph/utils/tensor_utils.h

/**
 * 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_