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()238 for (auto dimvec : input_profiles) { in TEST_P() local
35 std::vector<nvinfer1::Dims> dimvec(shape_vec.size()); in GetDimVec() local57 const std::vector<nvinfer1::Dims>& dimvec, std::vector<nvinfer1::Dims>* min, in SetImplicitBatchModeCompatibleProfile()263 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in InitProfiles() local
166 std::vector<int> dimvec(this->param_.dims.nbDims, 3 * i); in CreateTRTEngine() local
125 string DebugString(const std::vector<nvinfer1::Dims>& dimvec) { in DebugString()
72 std::vector<tensorflow::int64> dimvec(num_dims); in CreateTensor() local
635 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_AllocateHostTensor() local
572 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_NewTensorHandleFromDeviceMemory() local
647 std::vector<ssize_t> dimvec(dim_sizes.size()); in CreateTensor() local