diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-27 06:12:59 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 06:16:42 -0700 |
| commit | abf26356209cba1ba895a06d9ce55ad01dad7fc6 (patch) | |
| tree | 5ef1c907a30bf89d08ba241ef985b19938427420 /tensorflow/contrib/lite/kernels/select.cc | |
| parent | 19d8963bc0ea64e10ff08ad4e7cc76813a182196 (diff) | |
Update kernel evals to use new kernel signatures.
PiperOrigin-RevId: 214763814
Diffstat (limited to 'tensorflow/contrib/lite/kernels/select.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/select.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/kernels/select.cc b/tensorflow/contrib/lite/kernels/select.cc index 3959502d91..4780a86ee5 100644 --- a/tensorflow/contrib/lite/kernels/select.cc +++ b/tensorflow/contrib/lite/kernels/select.cc @@ -70,12 +70,12 @@ TfLiteStatus SelectEval(TfLiteContext* context, TfLiteNode* node) { bool is_rank_one = !HaveSameShapes(input_condition, input_x); -#define TF_LITE_SELECT(type, op) \ - reference_ops::op(GetTensorData<bool>(input_condition), \ - GetTensorDims(input_condition), \ - GetTensorData<type>(input_x), GetTensorDims(input_x), \ - GetTensorData<type>(input_y), GetTensorDims(input_y), \ - GetTensorData<type>(output), GetTensorDims(output)); +#define TF_LITE_SELECT(type, op) \ + reference_ops::op(GetTensorShape(input_condition), \ + GetTensorData<bool>(input_condition), \ + GetTensorShape(input_x), GetTensorData<type>(input_x), \ + GetTensorShape(input_y), GetTensorData<type>(input_y), \ + GetTensorShape(output), GetTensorData<type>(output)); #define TF_LITE_SWITCH(type, op) \ switch (type) { \ |