Searched refs:GetCustomOpOutputSize (Results 1 – 2 of 2) sorted by relevance
80 size_t GetCustomOpOutputSize(const CusOperatorPtr &cus_op);239 …size_t GetCustomOpOutputSize(const CusOperatorPtr &cus_op) { return impl_->GetCustomOpOutputSize(c… in GetCustomOpOutputSize() function
308 size_t OpAdapterImpl::GetCustomOpOutputSize(const CusOperatorPtr &cus_op) { in GetCustomOpOutputSize() function in mindspore::transform::OpAdapterImpl341 output_size = GetCustomOpOutputSize(std::dynamic_pointer_cast<CustomOperator>(op)); in UpdateMultiOutputDesc()