Searched refs:output_index_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unique.cc | 52 TfLiteTensor* output_index_tensor; in Prepare() local
54 &output_index_tensor)); in Prepare()
62 return context->ResizeTensor(context, output_index_tensor, in Prepare()
137 TfLiteTensor* output_index_tensor; in Eval() local
139 GetOutputSafe(context, node, 1, &output_index_tensor)); in Eval()
140 TF_LITE_ENSURE_EQ(context, NumElements(output_index_tensor), in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 3643 const TfLiteTensor& output_index_tensor = tensors[node->outputs->data[1]]; in VisitMediaPipeMaxPoolingNode() local
3644 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, output_index_tensor, in VisitMediaPipeMaxPoolingNode()
3647 CheckTensorNonDynamicAllocation(logging_context, output_index_tensor, in VisitMediaPipeMaxPoolingNode()
|