Searched refs:dims_0 (Results 1 – 1 of 1) sorted by relevance
671 const nvinfer1::Dims dims_0 = inputs.at(0).GetTrtDims(); in VerifyShapesMatch() local674 if (dim_i.nbDims != dims_0.nbDims) { in VerifyShapesMatch()678 for (size_t j = 0; j < dims_0.nbDims; ++j) { in VerifyShapesMatch()680 if (dim_i.d[j] == -1 || dims_0.d[j] == -1) continue; in VerifyShapesMatch()681 if (dim_i.d[j] != dims_0.d[j] && j != masked_dim) { in VerifyShapesMatch()