1 Star 0 Fork 204

zhangzhenghai/graphengine

forked from MindSpore/cann-headers
暂停
 
加入 Gitee
与超过 1400万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
data_slice_elementwise_impl.h 1.08 KB
一键复制 编辑 原始数据 按行查看 历史
胡彬 提交于 2023-03-11 09:11 +08:00 . keep pace with upgrade 20230308 run
/**
* Copyright (c) Huawei Technologies Co., Ltd. 2022. 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.
*/
#ifndef SLICE_DATASLICE_DATA_SLICE_ELEMENTWISE_IMPL_H_
#define SLICE_DATASLICE_DATA_SLICE_ELEMENTWISE_IMPL_H_
#include "slice/data_slice_infer_base.h"
namespace ge {
// Elementwise
class DataSliceElementwiseImpl : public DataSliceInferBase {
public:
Status InferAxisSlice(Operator &op, const AxisTypeInfo &slice_info,
const DataSliceType &out_data_slice, DataSliceType &in_data_slice) override;
};
}
#endif // SLICE_DATASLICE_DATA_SLICE_ELEMENTWISE_IMPL_H_
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C
1
https://gitee.com/majorzhang/graphengine.git
git@gitee.com:majorzhang/graphengine.git
majorzhang
graphengine
graphengine
21ac0e5d6b750466d986adc96e5c5bc1ee7e155d

搜索帮助