Searched refs:output_value_tensor (Results 1 – 2 of 2) sorted by relevance
3633 const TfLiteTensor& output_value_tensor = tensors[node->outputs->data[0]]; in VisitMediaPipeMaxPoolingNode() local3635 CheckTensorFloat32Type(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()3637 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()3640 CheckTensorNonDynamicAllocation(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()
4888 ITensorProxyPtr output_value_tensor = layer->getOutput(0); in ConvertTopK() local4890 params->outputs->push_back(TRT_TensorOrWeights(output_value_tensor)); in ConvertTopK()