Home
last modified time | relevance | path

Searched defs:trt_weights (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc406 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
421 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
438 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
Dconvert_nodes.cc472 nvinfer1::Weights trt_weights = weights.GetTrtWeights(); in CreateConstantLayer() local