| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -116,7 +116,7 @@ int BaseTensor<ACLTensor>::tensor_copy(void * mem,bool toTensor) | |||
| 116 | 116 | #endif //USE_PROFILING | |
| 117 | 117 | arm_compute::Window window; | |
| 118 | 118 | ACLTensor* tensor=this; | |
| 119 | - window.use_tensor_dimensions(tensor->info(), /* first_dimension =*/Window::DimY); // Iterate through the rows (not each element) | ||
| 119 | + window.use_tensor_dimensions(tensor->info()->tensor_shape(), /* first_dimension =*/Window::DimY); // Iterate through the rows (not each element) | ||
| 120 | 120 | int width = tensor->info()->tensor_shape()[0]; //->dimension(0); //window.x().end() - window.x().start(); // + 1; | |
| 121 | 121 | int height = tensor->info()->tensor_shape()[1]; //->dimension(1); //window.y().end() - window.y().start(); // + 1; | |
| 122 | 122 | int deepth = tensor->info()->tensor_shape()[2]; | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -115,13 +115,13 @@ void ACLBatchNormLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, | |||
| 115 | 115 | Forward_gpu(bottom, top); | |
| 116 | 116 | return; | |
| 117 | 117 | } | |
| 118 | + #ifdef USE_PROFILING | ||
| 119 | + logtime_util log_time(ACL_BN_INFO); | ||
| 120 | + #endif //USE_PROFILING | ||
| 118 | 121 | if (this->force_bypass_acl_path_||!this->use_global_stats_) { | |
| 119 | 122 | BatchNormLayer<Dtype>::Forward_cpu(bottom,top); | |
| 120 | 123 | return; | |
| 121 | 124 | } | |
| 122 | - #ifdef USE_PROFILING | ||
| 123 | - logtime_util log_time(ACL_BN_INFO); | ||
| 124 | - #endif //USE_PROFILING | ||
| 125 | 125 | const Dtype* bottom_data = bottom[0]->cpu_data(); | |
| 126 | 126 | Dtype* top_data = top[0]->mutable_cpu_data(); | |
| 127 | 127 | SetupACLLayer(bottom,top); | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -113,6 +113,7 @@ void ACLConcatLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, | |||
| 113 | 113 | ||
| 114 | 114 | template <typename Dtype> | |
| 115 | 115 | ACLConcatLayer<Dtype>::~ACLConcatLayer() { | |
| 116 | + if(this->force_bypass_acl_path_)return; | ||
| 116 | 117 | for (int i =0; i < cpu_vectors.size(); i ++) { | |
| 117 | 118 | delete cpu_vectors[i]; | |
| 118 | 119 | } | |
| Back | FazBrowse Home | New Git URL |
0 commit comments