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

change calibrator delegate mechanism to not mark graph immutable.

Browse Source 
PiperOrigin-RevId: 266925666
This commit is contained in:
Jian Li 2019-09-03 07:39:18 -07:00 committed by TensorFlower Gardener
parent a66d77fad7
commit 5987362c41
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/lite/tools/optimize/calibration/node_info_delegate.cc
View File

@ -33,11 +33,14 @@ TfLiteStatus NodeInfoDelegatePrepare(TfLiteContext* context,
} // namespace
TfLiteDelegate CreateNodeInfoDelegate(NodeInfoDelegateParams* params) {
return {/*data_ */ params,
/* Prepare */ NodeInfoDelegatePrepare,
/* CopyFromBufferHandle*/ nullptr,
/* CopyToBufferHandle*/ nullptr,
/* FreeBufferHandle*/ nullptr};
auto delegate = TfLiteDelegateCreate();
delegate.data_ = params;
delegate.Prepare = NodeInfoDelegatePrepare;
delegate.CopyFromBufferHandle = nullptr;
delegate.CopyToBufferHandle = nullptr;
delegate.FreeBufferHandle = nullptr;
delegate.flags = kTfLiteDelegateFlagsAllowDynamicTensors;
return delegate;
}
TfLiteStatus NodeInfoDelegateObserver::OnDelegatePrepareCalled(