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

Fixed bias alignment.

Browse Source 
PiperOrigin-RevId: 342278915
Change-Id: I6dc5edcc1c4ac4f932233e2f34470f272638ceea
This commit is contained in:
Raman Sarokin 2020-11-13 09:36:59 -08:00 committed by TensorFlower Gardener
parent 691a4e830f
commit 3ad7998055
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/delegates/gpu/metal/kernels/fully_connected.cc
View File

@ -183,7 +183,7 @@ std::vector<ComputeTaskDescriptorPtr> FullyConnected(
: DataType::FLOAT16;
bias_desc.element_size = 4;
bias_desc.data = GetByteBufferConvertedResized(
attr.bias.data, options.storage_precision, attr.weights.shape.o);
attr.bias.data, options.storage_precision, dst_channels_aligned);
bias_desc.size = bias_desc.data.size();
desc->args.AddObject(