Searched refs:SetCustomOpInput (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.h | 68 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &input);
71 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handle);
200 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &input) { in SetCustomOpInput() function
201 return impl_->SetCustomOpInput(op, index, input); in SetCustomOpInput()
212 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handle) { in SetCustomOpInput() function
213 return impl_->SetCustomOpInput(op, index, handle); in SetCustomOpInput()
|
| D | op_adapter.cc | 123 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &inpu… in SetCustomOpInput() function in mindspore::transform::OpAdapterImpl
154 return static_cast<int>(SetCustomOpInput(cus_op, index, input)); in setInput()
160 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OutHandler &handl… in SetCustomOpInput() function in mindspore::transform::OpAdapterImpl
202 return static_cast<int>(SetCustomOpInput(cus_op, index, handle)); in setInput()
|