Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3051 const DataType converted_dtype = in TfTensorToTrtWeights() local
3057 TF_RETURN_IF_ERROR(ConvertDType(converted_dtype, &trt_dtype)); in TfTensorToTrtWeights()
3061 *weights = TRT_ShapedWeights(converted_dtype); in TfTensorToTrtWeights()
3067 *weights = weight_store->GetTempWeights(converted_dtype, weight_dims); in TfTensorToTrtWeights()
3071 if (converted_dtype == dtype) { in TfTensorToTrtWeights()