Searched refs:CheckTensorShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 643 static TfLiteStatus CheckTensorShape(TfLiteContext* context, in CheckTensorShape() function in tflite::xnnpack::__anon1357bc120111::Subgraph
686 static TfLiteStatus CheckTensorShape(TfLiteContext* context, in CheckTensorShape() function in tflite::xnnpack::__anon1357bc120111::Subgraph
690 return CheckTensorShape(context, tensor, expected_num_dims, in CheckTensorShape()
1237 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, input_tensor, 4, in VisitConv2DNode()
1245 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, filter_tensor, 4, in VisitConv2DNode()
1262 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, bias_tensor, 1, in VisitConv2DNode()
1272 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, output_tensor, 4, in VisitConv2DNode()
1332 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, input_tensor, 4, in VisitDepthwiseConv2DNode()
1340 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, filter_tensor, 4, in VisitDepthwiseConv2DNode()
1357 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, bias_tensor, 1, in VisitDepthwiseConv2DNode()
[all …]
|