Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h237 TRT_ShapedWeights GetTempWeights(DataType type, const nvinfer1::Dims& dims);
241 TRT_ShapedWeights GetTempWeights(const TRT_ShapedWeights& weights) { in GetTempWeights() function
242 return GetTempWeights(weights.type_, weights.shape_); in GetTempWeights()
Dconvert_nodes.cc480 params->weight_store->GetTempWeights(dtype, broadcastable_dims); in CreateBroadcastableScalarConstant()
872 TRT_ShapedWeights TrtWeightStore::GetTempWeights(DataType type, in GetTempWeights() function in tensorflow::tensorrt::convert::TrtWeightStore
1484 store->GetTempWeights(dtype_new, weights_src.shape_); in ConvertFP32ToFP16()
1693 params->weight_store->GetTempWeights(weights); in BinaryTensorOpWeight()
1728 params->weight_store->GetTempWeights(weights); in BinaryTensorOpWeight()
1847 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()
3067 *weights = weight_store->GetTempWeights(converted_dtype, weight_dims); in TfTensorToTrtWeights()
3644 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()
3646 params->weight_store->GetTempWeights(*ptr_shape_weights); in ConvertFusedBatchNorm()
3798 weights = params->weight_store->GetTempWeights(weights_raw); in ConvertMatMulHelper()
Dconvert_nodes_test.cc303 store.GetTempWeights(DT_FLOAT, GetTestDims({2, 5})); in TEST()
751 weight_store_->GetTempWeights(DT_FLOAT, GetTestDims({2, 3, 5})); in TEST_F()
805 weight_store->GetTempWeights(DataTypeToEnum<T>::v(), GetTestDims({2, 3})); in TestGetWeightRange()
977 weight_store_->GetTempWeights(dtype, GetTestDims({2, 3, 5})); in TEST_F()
1224 weights = converter_->weight_store_.GetTempWeights(dtype, trt_dims); in AddTestWeights()