experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change the estimation of cost in parallelFor

Browse Source
This commit is contained in:
leslie-fang-intel 2019-11-22 09:32:22 +08:00
parent d7757cbe27
commit 4f85ec0291
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/kernels/resize_nearest_neighbor_op.cc
View File

@ -165,12 +165,12 @@ struct ResizeNearestNeighbor<CPUDevice, T, half_pixel_centers, align_corners> {
}
};
Eigen::Index N = batch_size * out_height * out_width;
const int input_bytes =
batch_size * in_height * in_width * channels * sizeof(T);
const int output_bytes = N * channels * sizeof(T);
const int input_bytes = channels * sizeof(T);
const int output_bytes = channels * sizeof(T);
const int compute_cycles = (Eigen::TensorOpCost::ModCost<T>() * 2 +
Eigen::TensorOpCost::DivCost<T>() * 5) *
N;
Eigen::TensorOpCost::DivCost<T>() * 3 +
Eigen::TensorOpCost::AddCost<T>() * 2 +
Eigen::TensorOpCost::MulCost<T>() * 2);
const Eigen::TensorOpCost cost(input_bytes, output_bytes, compute_cycles);
d.parallelFor(N, cost, ParallelResize);
return true;