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

Minor changes

Browse Source
This commit is contained in:
Kaixi Hou 2020-05-11 10:46:01 -07:00
parent b9579f96bd
commit 0975574df3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/conv_2d_gpu.h
View File

@ -210,7 +210,7 @@ __global__ void ShuffleInTensor3Simple(int nthreads,
}
}
constexpr int kUnroll = 4;
static constexpr int kUnroll = 4;
template <typename T, int sp0, int sp1, int sp2, bool conjugate = false>
__global__ void ShuffleInTensor3SimpleVector(int nthreads,
@ -246,7 +246,7 @@ __global__ void ShuffleInTensor3SimpleVector(int nthreads,
*out = *reinterpret_cast<float2*>(buf);
}
for(; output_index < nthreads; output_index++) {
for (; output_index < nthreads; ++output_index) {
Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
Index<3> input_tensor_index;