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

iOS TFLite delegate: align uniform buffer for concat operation - remove xcode Metal validator warning.

Browse Source 
PiperOrigin-RevId: 286101150
Change-Id: Ia5bae78b1f68f8f35cfe75457f3097ae954d58e6
This commit is contained in:
A. Unique TensorFlower 2019-12-17 18:07:29 -08:00 committed by TensorFlower Gardener
parent 285188f415
commit c69bd15c41
1 changed files with 2 additions and 1 deletions

3
tensorflow/lite/delegates/gpu/metal/kernels/concat.cc
View File

@ -329,7 +329,8 @@ std::vector<ComputeTaskDescriptorPtr> ConcatY(
[output_id](const std::map<ValueId, BHWC>& buffers) {
const auto& dimension = buffers.find(output_id)->second;
std::vector<int> uniform_params{dimension.w, dimension.h,
IntegralDivideRoundUp(dimension.c, 4)};
IntegralDivideRoundUp(dimension.c, 4),
/*padding=*/0};
return GetByteBuffer(uniform_params);
}},
};