Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5226 const TRT_ShapedWeights* ptr_shape_weights = nullptr; in ConvertFusedBatchNorm() local
5229 ptr_shape_weights = &(inputs.at(i).weights()); in ConvertFusedBatchNorm()
5240 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()
5242 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()