Searched defs:dimvec (Results 1 – 6 of 6) sorted by relevance
31 std::vector<nvinfer1::Dims> dimvec(shape_vec.size()); in GetDimVec() local51 const std::vector<nvinfer1::Dims>& dimvec, std::vector<nvinfer1::Dims>* min, in SetImplicitBatchModeCompatibleProfile()64 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in ImplicitBatchModeCompatibleStrategy() local
34 std::vector<TensorShape> DimVecToShapeVec(std::vector<nvinfer1::Dims3> dimvec) { in DimVecToShapeVec()191 for (auto dimvec : input_profiles) { in TEST_F() local
139 string DebugString(const std::vector<nvinfer1::Dims>& dimvec) { in DebugString()
72 std::vector<tensorflow::int64> dimvec(num_dims); in CreateTensor() local
579 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_AllocateHostTensor() local
578 std::vector<tensorflow::int64> dimvec(num_dims); in TFE_NewTensorHandleFromDeviceMemory() local