File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1-
2- #ifndef WIN32_LEAN_AND_MEAN
1+ #ifndef WIN32_LEAN_AND_MEAN
32#define WIN32_LEAN_AND_MEAN
43#endif
5- #include < windows.h>
64#include < psapi.h>
5+ #include < windows.h>
76#pragma comment(lib, "psapi.lib")
87#include < crtdbg.h>
98#include < algorithm>
Original file line number Diff line number Diff line change @@ -1142,5 +1142,6 @@ int print_time_stats(Graph& graph) {
11421142 return sum;
11431143#else
11441144 (void )graph;
1145+ return 0 ;
11451146#endif
11461147}
Original file line number Diff line number Diff line change @@ -318,11 +318,9 @@ class Graph {
318318
319319 for (size_t i = 0 ; i < traversal.size (); ++i) {
320320 int current_layer = traversal[i];
321-
322321#ifdef ENABLE_STATISTIC_TIME
323322 auto start = std::chrono::high_resolution_clock::now ();
324323#endif
325-
326324 if (i != 0 ) {
327325 inten_.clear ();
328326
@@ -345,19 +343,15 @@ class Graph {
345343 }
346344
347345 it->second .count_used_ten --;
348-
349- // Åñëè òåíçîð áîëüøå íèêîìó íå íóæåí - óäàëÿåì
350346 if (it->second .count_used_ten < 1 ) {
351347 branch_map_.erase (it);
352348 }
353349 }
354350 }
355351 }
356-
357352 if (outten_.empty ()) {
358353 outten_.resize (1 );
359354 }
360-
361355 layers_[current_layer]->run (inten_, outten_, options);
362356
363357#ifdef ENABLE_STATISTIC_TENSORS
@@ -405,9 +399,7 @@ class Graph {
405399 }
406400 new_branch.distribution = dis;
407401 }
408-
409402 branch_map_[current_layer] = std::move (new_branch);
410-
411403 if (outtenres_ && current_layer == end_ &&
412404 !branch_map_[current_layer].give_for_all .empty () &&
413405 countinout[current_layer].second == 0 ) {
@@ -441,11 +433,6 @@ class Graph {
441433 }
442434#endif
443435 }
444-
445- // Ôèíàëüíàÿ î÷èñòêà
446- branch_map_.clear ();
447- inten_.clear ();
448- outten_.clear ();
449436 }
450437
451438 void setOutput (const std::shared_ptr<Layer>& layer, Tensor& vec) {
You can’t perform that action at this time.
0 commit comments