Searched refs:CreateConstantLayer (Results 1 – 3 of 3) sorted by relevance
594 nvinfer1::ITensor* CreateConstantLayer(const TRT_ShapedWeights& weights,
470 nvinfer1::ITensor* Converter::CreateConstantLayer( in CreateConstantLayer() function in tensorflow::tensorrt::convert::Converter498 *tensor = params->converter->CreateConstantLayer(weights, dims);1747 *tensor = converter->CreateConstantLayer(input.weights(), dims); in PrepareTensorForShape()2549 auto output = params->converter->CreateConstantLayer(weight, output_dims); in ConvertShape()5368 params_tensor = params->converter->CreateConstantLayer( in ConvertGather()5528 return params->converter->CreateConstantLayer(weights, weights.shape_); in ConvertMatMulHelper()6207 params->converter->CreateConstantLayer(input.weights(), dims)); in ConvertAddN()
1249 TEST_F(ConverterTest, CreateConstantLayer) { in TEST_F() argument1254 converter_->CreateConstantLayer(weights, GetTestDims({3, 10})); in TEST_F()