Searched refs:layer_consumers (Results 1 – 1 of 1) sorted by relevance
1844 std::map<nvinfer1::ILayer*, std::vector<nvinfer1::ILayer*>> layer_consumers; in MaybeApplyQuantizationRanges() local1849 layer_consumers[layer] = {}; in MaybeApplyQuantizationRanges()1860 layer_consumers[layer] = {}; in MaybeApplyQuantizationRanges()1865 auto consumed_layer = layer_consumers.find(input_layer->second); in MaybeApplyQuantizationRanges()1866 if (consumed_layer != layer_consumers.end()) { in MaybeApplyQuantizationRanges()1920 (index < last_matcher && layer_consumers[layer].size() != 1)) { in MaybeApplyQuantizationRanges()1926 layer = layer_consumers[layer].front(); in MaybeApplyQuantizationRanges()