Searched refs:tensor_layer (Results 1 – 1 of 1) sorted by relevance
1845 std::map<nvinfer1::ITensor*, nvinfer1::ILayer*> tensor_layer; in MaybeApplyQuantizationRanges() local1854 tensor_layer[layer->getOutput(j)] = layer; in MaybeApplyQuantizationRanges()1863 auto input_layer = tensor_layer.find(input_tensor); in MaybeApplyQuantizationRanges()1864 if (input_layer != tensor_layer.end()) { in MaybeApplyQuantizationRanges()1957 auto layer = tensor_layer.find(tensor); in MaybeApplyQuantizationRanges()1961 if (layer != tensor_layer.end() && in MaybeApplyQuantizationRanges()