Searched refs:GetTrtDims (Results 1 – 3 of 3) sorted by relevance
399 const nvinfer1::Dims input_dims = input.GetTrtDims(); in GetTrtBroadcastShape()432 std::max(operand_l.GetTrtDims().nbDims + in GetTrtBroadcastShape()434 operand_r.GetTrtDims().nbDims + in GetTrtBroadcastShape()671 const nvinfer1::Dims dims_0 = inputs.at(0).GetTrtDims(); in VerifyShapesMatch()673 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch()871 nvinfer1::Dims TRT_TensorOrWeights::GetTrtDims() const { in GetTrtDims() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights1709 const nvinfer1::Dims input_dims = input.GetTrtDims(); in PrepareTensorForShape()2540 if (HasStaticShape(inputs.at(0).GetTrtDims())) { in ConvertShape()2542 nvinfer1::Dims input_dims = inputs.at(0).GetTrtDims(); in ConvertShape()2593 const nvinfer1::Dims input_nonbatch_dims = input_tensor.GetTrtDims(); in ConvertReshape()[all …]
302 nvinfer1::Dims GetTrtDims() const;
489 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()508 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()522 ExpectTrtDimsEqualsArray({}, ptr->GetTrtDims()); in TEST()613 ExpectTrtDimsEqualsArray({non_batch_dim}, output.GetTrtDims()); in TEST_F()