Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h357 weight_store(arg_weight_store) {} in OpConverterParams()
364 TrtWeightStore* weight_store; member
Dconvert_nodes.cc480 params->weight_store->GetTempWeights(dtype, broadcastable_dims); in CreateBroadcastableScalarConstant()
1671 weights = ConvertFP32ToFP16(params->weight_store, weights); in BinaryTensorOpWeight()
1693 params->weight_store->GetTempWeights(weights); in BinaryTensorOpWeight()
1728 params->weight_store->GetTempWeights(weights); in BinaryTensorOpWeight()
1841 weights_rsck = ConvertFP32ToFP16(params->weight_store, weights_rsck); in ConvertConv2DHelper()
1847 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()
2984 weights = ConvertFP32ToFP16(params->weight_store, weights); in ConvertBiasAdd()
3041 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights() argument
3067 *weights = weight_store->GetTempWeights(converted_dtype, weight_dims); in TfTensorToTrtWeights()
3126 TfTensorToTrtWeights(tensor, params->weight_store, &weights)); in ConvertConst()
[all …]
Dconvert_nodes_test.cc803 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange() argument
805 weight_store->GetTempWeights(DataTypeToEnum<T>::v(), GetTestDims({2, 3})); in TestGetWeightRange()