Searched refs:cus_op (Results 1 – 2 of 2) sorted by relevance
153 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in setInput() local154 return static_cast<int>(SetCustomOpInput(cus_op, index, input)); in setInput()201 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in setInput() local202 return static_cast<int>(SetCustomOpInput(cus_op, index, handle)); in setInput()294 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in UpdateSingleOutputDesc() local295 MS_EXCEPTION_IF_NULL(cus_op); in UpdateSingleOutputDesc()297 (void)cus_op->UpdateOutputDesc(output_map[0], *desc); in UpdateSingleOutputDesc()308 size_t OpAdapterImpl::GetCustomOpOutputSize(const CusOperatorPtr &cus_op) { in GetCustomOpOutputSize() argument309 MS_EXCEPTION_IF_NULL(cus_op); in GetCustomOpOutputSize()310 if (cus_output_map_->find(cus_op->GetOpType()) == cus_output_map_->end()) { in GetCustomOpOutputSize()[all …]
80 size_t GetCustomOpOutputSize(const CusOperatorPtr &cus_op);239 …ize_t GetCustomOpOutputSize(const CusOperatorPtr &cus_op) { return impl_->GetCustomOpOutputSize(cu… in GetCustomOpOutputSize() argument