Searched refs:trt_weights (Results 1 – 2 of 2) sorted by relevance
406 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local407 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()408 EXPECT_EQ(nullptr, trt_weights.values); in TEST()409 EXPECT_EQ(0, trt_weights.count); in TEST()421 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local422 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()423 EXPECT_EQ(nullptr, trt_weights.values); in TEST()424 EXPECT_EQ(0, trt_weights.count); in TEST()438 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local439 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()[all …]
472 nvinfer1::Weights trt_weights = weights.GetTrtWeights(); in CreateConstantLayer() local473 nvinfer1::IConstantLayer* layer = network()->addConstant(dims, trt_weights); in CreateConstantLayer()484 trt_tensor->setType(trt_weights.type); in CreateConstantLayer()