Beispiel #1
0
        private void getTestingDataFn(long nBatchSize, IntPtr hVal)
        {
            if (OnGetTestingDataBatch == null)
            {
                return;
            }

            DataBatchArgs args = new DataBatchArgs(nBatchSize, m_rgExpectedInputs, m_rgExpectedOutputs);

            OnGetTestingDataBatch(this, args);
            handleGetDataFn(args, hVal);
        }
Beispiel #2
0
        private void handleGetDataFn(DataBatchArgs args, IntPtr hcol)
        {
            OrtValueCollection col = new OrtValueCollection(hcol);

            for (int i = 0; i < args.Values.Count; i++)
            {
                MemoryHandle?memHandle;
                OrtValue     val = args.Values[i].ToOrtValue(out memHandle);

                if (memHandle.HasValue)
                {
                    m_rgCleanUpList.Add(memHandle);
                }

                m_rgCleanUpList.Add(val);

                col.SetAt(i, val, args.Values[i].Name);
            }
        }