Home
last modified time | relevance | path

Searched refs:tensor_layer (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1845 std::map<nvinfer1::ITensor*, nvinfer1::ILayer*> tensor_layer; in MaybeApplyQuantizationRanges() local
1854 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()