Searched refs:HasStaticShape (Results 1 – 3 of 3) sorted by relevance
114 inline bool HasStaticShape(const nvinfer1::Dims& dims) { in HasStaticShape() function122 inline bool HasStaticShape(std::vector<int> dims) { in HasStaticShape() function
604 if (!HasStaticShape(dims)) return -1; in TrtTensorDimsNumElements()620 if (!HasStaticShape(lhs) || !HasStaticShape(rhs)) return false; in AreDimsStaticWithSameSize()626 if (!HasStaticShape(lhs) || !HasStaticShape(rhs)) return false; in AreDimsStaticWithDifferentSize()1723 if (input.is_weights() && !HasStaticShape(dims)) { in PrepareTensorForShape()2540 if (HasStaticShape(inputs.at(0).GetTrtDims())) { in ConvertShape()2662 if (!params->use_implicit_batch && !HasStaticShape(input_dims)) { in ConvertExpandDims()2764 if (!params->use_implicit_batch && !HasStaticShape(*input_dims)) { in SqueezeTensor()2799 if (params->use_implicit_batch || !HasStaticShape(dims)) { in ConvertSqueeze()4762 if (HasStaticShape(inputs.at(i).GetTrtDims())) { in ConvertPack()4783 !HasStaticShape(dims)) { in ConvertPack()[all …]
1546 if (!converter_->use_implicit_batch() || HasStaticShape(trt_dims)) { in AddTestTensorWithTFDims()1567 if (HasStaticShape(dims)) { in AddTestTensor()