Home
last modified time | relevance | path

Searched refs:OpAdapterImpl (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc23 bool OpAdapterImpl::IsCustomOp(const OperatorPtr &op) { in IsCustomOp()
32 Status OpAdapterImpl::GenerateCustomOpInputMap(const CusOperatorPtr &op, const PrimitivePtr &prim) { in GenerateCustomOpInputMap()
56 Status OpAdapterImpl::GenerateCustomOpOutputMap(const CusOperatorPtr &op, const PrimitivePtr &prim)… in GenerateCustomOpOutputMap()
81 OperatorPtr OpAdapterImpl::GenerateCustomOp(const AnfNodePtr anf) { in GenerateCustomOp()
108 Status OpAdapterImpl::SetOpSubgraphFunc(const OperatorPtr &op, int index, in SetOpSubgraphFunc()
123 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &inpu… in SetCustomOpInput()
140 Status OpAdapterImpl::SetNormalOpInput(const OperatorPtr &op, int index, const OperatorPtr &input) { in SetNormalOpInput()
151 int OpAdapterImpl::setInput(const OperatorPtr &op, int index, const OperatorPtr &input) { in setInput()
160 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handl… in SetCustomOpInput()
182 Status OpAdapterImpl::SetNormalOpInput(const OperatorPtr &op, int index, const OutHandler &handle) { in SetNormalOpInput()
[all …]
Dop_adapter.h29 class OpAdapterImpl {
31 OpAdapterImpl(const std::unordered_map<int, InputDesc> &input_map, in OpAdapterImpl() function
62 ~OpAdapterImpl() {} in ~OpAdapterImpl()
118 … : impl_(std::make_shared<OpAdapterImpl>(input_map_, dyn_input_map_, output_map_, dyn_output_map_, in OpAdapter()
125 … impl_(std::make_shared<OpAdapterImpl>(input_map_, dyn_input_map_, output_map_, dyn_output_map_, in OpAdapter()
439 const std::shared_ptr<OpAdapterImpl> impl_;