Home
last modified time | relevance | path

Searched refs:CusOperatorPtr (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.h64 Status GenerateCustomOpInputMap(const CusOperatorPtr &op, const PrimitivePtr &prim);
65 Status GenerateCustomOpOutputMap(const CusOperatorPtr &op, const PrimitivePtr &prim);
68 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &input);
71 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handle);
80 size_t GetCustomOpOutputSize(const CusOperatorPtr &cus_op);
87 …void UpdateCustomOpInputDesc(const CusOperatorPtr &op, const AnfNodePtr &node, const std::string f…
92 int SetCustomOpAttr(const CusOperatorPtr &op, const PrimitivePtr &prim);
134 Status GenerateCustomOpInputMap(const CusOperatorPtr &op, const PrimitivePtr &prim) { in GenerateCustomOpInputMap()
138 Status GenerateCustomOpOutputMap(const CusOperatorPtr &op, const PrimitivePtr &prim) { in GenerateCustomOpOutputMap()
200 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &input) { in SetCustomOpInput()
[all …]
Dop_adapter.cc32 Status OpAdapterImpl::GenerateCustomOpInputMap(const CusOperatorPtr &op, const PrimitivePtr &prim) { in GenerateCustomOpInputMap()
56 Status OpAdapterImpl::GenerateCustomOpOutputMap(const CusOperatorPtr &op, const PrimitivePtr &prim)… in GenerateCustomOpOutputMap()
123 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &inpu… in SetCustomOpInput()
160 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handl… in SetCustomOpInput()
308 size_t OpAdapterImpl::GetCustomOpOutputSize(const CusOperatorPtr &cus_op) { in GetCustomOpOutputSize()
425 void OpAdapterImpl::UpdateCustomOpInputDesc(const CusOperatorPtr &op, const AnfNodePtr &node, in UpdateCustomOpInputDesc()
509 int OpAdapterImpl::SetCustomOpAttr(const CusOperatorPtr &op, const PrimitivePtr &prim) { in SetCustomOpAttr()
Dop_adapter_base.h65 using CusOperatorPtr = std::shared_ptr<ge::CustomOperator>; variable