Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc274 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
275 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()
276 EXPECT_EQ(nullptr, trt_weights.values); in TEST()
277 EXPECT_EQ(0, trt_weights.count); in TEST()
289 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
290 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()
291 EXPECT_EQ(nullptr, trt_weights.values); in TEST()
292 EXPECT_EQ(0, trt_weights.count); in TEST()
306 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST() local
307 EXPECT_EQ(nvinfer1::DataType::kFLOAT, trt_weights.type); in TEST()
[all …]
Dconvert_nodes.cc447 nvinfer1::Weights trt_weights = weights.GetTrtWeights(); in CreateConstantLayer() local
448 nvinfer1::IConstantLayer* layer = network()->addConstant(dims, trt_weights); in CreateConstantLayer()
450 const nvinfer1::DataType trt_dtype = trt_weights.type; in CreateConstantLayer()