Searched refs:GetTrtDims (Results 1 – 3 of 3) sorted by relevance
399 const nvinfer1::Dims input_dims = input.GetTrtDims(); in GetTrtBroadcastShape()424 std::max(operand_l.GetTrtDims().nbDims + (operand_l.is_tensor() ? 1 : 0), in GetTrtBroadcastShape()425 operand_r.GetTrtDims().nbDims + (operand_r.is_tensor() ? 1 : 0)); in GetTrtBroadcastShape()745 nvinfer1::Dims TRT_TensorOrWeights::GetTrtDims() const { in GetTrtDims() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights1239 TrtDimsNumElements(input.GetTrtDims()) != TrtDimsNumElements(dims)) { in PrepareTensorForShape()1241 DebugString(input.GetTrtDims()), " vs ", in PrepareTensorForShape()1250 if (DimsEqual(input.GetTrtDims(), dims)) { in PrepareTensorForShape()2129 const nvinfer1::Dims input_dims = input_tensor.GetTrtDims(); in ConvertReshape()2186 const nvinfer1::Dims dims = input_tensor.GetTrtDims(); in ConvertExpandDims()2238 const nvinfer1::Dims dims = input_tensor.GetTrtDims(); in ConvertSqueeze()[all …]
296 nvinfer1::Dims GetTrtDims() const;
357 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()376 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()390 ExpectTrtDimsEqualsArray({}, ptr->GetTrtDims()); in TEST()475 ExpectTrtDimsEqualsArray({non_batch_dim}, output.GetTrtDims()); in TEST_F()