Searched defs:dimvec (Results 1 – 8 of 8) sorted by relevance
35 std::vector<nvinfer1::Dims3> dimvec, in DimVecToShapeVec()110 void CheckProfile(const std::vector<nvinfer1::Dims3>& dimvec, in CheckProfile()243 for (auto dimvec : input_profiles) { in TEST_P() local
37 std::vector<nvinfer1::Dims> dimvec(shape_vec.size()); in GetDimVec() local59 const std::vector<nvinfer1::Dims>& dimvec, std::vector<nvinfer1::Dims>* min, in SetImplicitBatchModeCompatibleProfile()274 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in InitProfiles() local331 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in InitCalibProfile() local
96 string DebugString(const std::vector<nvinfer1::Dims>& dimvec) { in DebugString()
172 std::vector<int> dimvec(this->param_.dims.nbDims, 3 * i); in CreateTRTEngine() local
72 std::vector<int64_t> dimvec(num_dims); in CreateTensor() local
635 std::vector<int64_t> dimvec(num_dims); in TFE_AllocateHostTensor() local
574 std::vector<int64_t> dimvec(num_dims); in TFE_NewTensorHandleFromDeviceMemory() local
688 std::vector<Index> dimvec(dim_sizes.size()); in CreateTensor() local