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 } // namespace
TfLiteDelegate CreateNodeInfoDelegate(NodeInfoDelegateParams* params) { TfLiteDelegate CreateNodeInfoDelegate(NodeInfoDelegateParams* params) {
return {/*data_ */ params, auto delegate = TfLiteDelegateCreate();
/* Prepare */ NodeInfoDelegatePrepare, delegate.data_ = params;
/* CopyFromBufferHandle*/ nullptr, delegate.Prepare = NodeInfoDelegatePrepare;
/* CopyToBufferHandle*/ nullptr, delegate.CopyFromBufferHandle = nullptr;
/* FreeBufferHandle*/ nullptr}; delegate.CopyToBufferHandle = nullptr;
delegate.FreeBufferHandle = nullptr;
delegate.flags = kTfLiteDelegateFlagsAllowDynamicTensors;
return delegate;
} }
TfLiteStatus NodeInfoDelegateObserver::OnDelegatePrepareCalled( TfLiteStatus NodeInfoDelegateObserver::OnDelegatePrepareCalled(