Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3065 nvinfer1::Dims weight_dims; in TfTensorToTrtWeights() local
3066 GetTensorDimsWithProtoShape(tensor, &weight_dims); in TfTensorToTrtWeights()
3067 *weights = weight_store->GetTempWeights(converted_dtype, weight_dims); in TfTensorToTrtWeights()