Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1481 const TRT_ShapedWeights& weights_src) { in ConvertFP32ToFP16() argument
1484 store->GetTempWeights(dtype_new, weights_src.shape_); in ConvertFP32ToFP16()
1485 const float* src = static_cast<const float*>(weights_src.GetValues()); in ConvertFP32ToFP16()
1488 for (int64_t i = 0; i < weights_src.count(); i++) { in ConvertFP32ToFP16()