Searched refs:CreateConstantLayer (Results 1 – 3 of 3) sorted by relevance
514 nvinfer1::ITensor* CreateConstantLayer(const TRT_ShapedWeights& weights,
445 nvinfer1::ITensor* Converter::CreateConstantLayer( in CreateConstantLayer() function in tensorflow::tensorrt::convert::Converter493 *tensor = params->converter->CreateConstantLayer(weights, broadcastable_dims); in CreateBroadcastableScalarConstant()1262 *tensor = CreateConstantLayer(input.weights(), dims); in PrepareTensorForShape()
974 TEST_F(ConverterTest, CreateConstantLayer) { in TEST_F() argument979 converter_->CreateConstantLayer(weights, GetTestDims({3, 10})); in TEST_F()