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

internal code change 

PiperOrigin-RevId: 305495596
Change-Id: I35d379bb956c2b786e288209b8e6b366698660b1
This commit is contained in:
Taehee Jeong 2020-04-08 09:42:28 -07:00 committed by TensorFlower Gardener
parent 3009d8c3ff
commit 8c0ac31fa8
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/lite/experimental/delegates/coreml/builders/fully_connected_op_builder.cc
View File

@ -53,7 +53,7 @@ void FullyConnectedOpBuilder::FillCoreMLWeights() {
layer_->mutable_innerproduct()->set_outputchannels(weights_->dims->data[0]); layer_->mutable_innerproduct()->set_outputchannels(weights_->dims->data[0]);
const float* weights_data = GetTensorData<float>(weights_); const float* weights_data = GetTensorData<float>(weights_);
std::copy(weights_data, weights_data + NumElements(weights_), std::copy(weights_data, weights_data + NumElements(weights_),
proto2::RepeatedFieldBackInserter(layer_->mutable_innerproduct() google::protobuf::RepeatedFieldBackInserter(layer_->mutable_innerproduct()
->mutable_weights() ->mutable_weights()
->mutable_floatvalue())); ->mutable_floatvalue()));
} }
@ -63,7 +63,7 @@ void FullyConnectedOpBuilder::FillCoreMLBias() {
layer_->mutable_innerproduct()->set_hasbias(true); layer_->mutable_innerproduct()->set_hasbias(true);
const float* bias_data = GetTensorData<float>(bias_); const float* bias_data = GetTensorData<float>(bias_);
std::copy(bias_data, bias_data + NumElements(bias_), std::copy(bias_data, bias_data + NumElements(bias_),
proto2::RepeatedFieldBackInserter(layer_->mutable_innerproduct() google::protobuf::RepeatedFieldBackInserter(layer_->mutable_innerproduct()
->mutable_bias() ->mutable_bias()
->mutable_floatvalue())); ->mutable_floatvalue()));
} }