Searched refs:output_value_tensor (Results 1 – 2 of 2) sorted by relevance
2125 const TfLiteTensor& output_value_tensor = tensors[node->outputs->data[0]]; in VisitMediaPipeMaxPoolingNode() local2127 CheckTensorFloatType(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()2129 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()2132 CheckTensorNonDynamicAllocation(logging_context, output_value_tensor, in VisitMediaPipeMaxPoolingNode()
5771 nvinfer1::ITensor* output_value_tensor = layer->getOutput(0); in ConvertTopK() local5773 params->outputs->push_back(TRT_TensorOrWeights(output_value_tensor)); in ConvertTopK()