diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/logical.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/logical.cc | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tensorflow/contrib/lite/kernels/logical.cc b/tensorflow/contrib/lite/kernels/logical.cc index 87c2fee667..c71f3b4701 100644 --- a/tensorflow/contrib/lite/kernels/logical.cc +++ b/tensorflow/contrib/lite/kernels/logical.cc @@ -86,14 +86,14 @@ TfLiteStatus LogicalImpl(TfLiteContext* context, TfLiteNode* node, TfLiteTensor* output = GetOutput(context, node, kOutputTensor); if (data->requires_broadcast) { - reference_ops::BroadcastLogical( - GetTensorData<bool>(input1), GetTensorDims(input1), - GetTensorData<bool>(input2), GetTensorDims(input2), - GetTensorData<bool>(output), GetTensorDims(output), func); + reference_ops::BroadcastLogical4DSlow( + GetTensorShape(input1), GetTensorData<bool>(input1), + GetTensorShape(input2), GetTensorData<bool>(input2), + GetTensorShape(output), GetTensorData<bool>(output), func); } else { - reference_ops::Logical(GetTensorData<bool>(input1), GetTensorDims(input1), - GetTensorData<bool>(input2), GetTensorDims(input2), - GetTensorData<bool>(output), GetTensorDims(output), + reference_ops::Logical(GetTensorShape(input1), GetTensorData<bool>(input1), + GetTensorShape(input2), GetTensorData<bool>(input2), + GetTensorShape(output), GetTensorData<bool>(output), func); } |