Searched refs:TrtWeightDimsNumElements (Results 1 – 3 of 3) sorted by relevance
166 int64_t TrtWeightDimsNumElements(const nvinfer1::Dims& dims);
594 int64_t TrtWeightDimsNumElements(const nvinfer1::Dims& dims) { in TrtWeightDimsNumElements() function613 return TrtWeightDimsNumElements(lhs) == TrtWeightDimsNumElements(rhs); in DimsHaveSameSize()702 return TrtWeightDimsNumElements(shape_); in count()
1586 const int64_t num_elements = TrtWeightDimsNumElements(trt_dims); in AddTestWeights()