Searched refs:weight_store (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.h | 357 weight_store(arg_weight_store) {} in OpConverterParams() 364 TrtWeightStore* weight_store; member
|
D | convert_nodes.cc | 480 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 …]
|
D | convert_nodes_test.cc | 803 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange() argument 805 weight_store->GetTempWeights(DataTypeToEnum<T>::v(), GetTestDims({2, 3})); in TestGetWeightRange()
|