Searched refs:TrtDimsNumElements (Results 1 – 3 of 3) sorted by relevance
171 int64_t TrtDimsNumElements(const nvinfer1::Dims& dims);
568 int64_t TrtDimsNumElements(const nvinfer1::Dims& dims) { in TrtDimsNumElements() function622 int64_t TRT_ShapedWeights::count() const { return TrtDimsNumElements(shape_); } in count()1239 TrtDimsNumElements(input.GetTrtDims()) != TrtDimsNumElements(dims)) { in PrepareTensorForShape()2143 TrtDimsNumElements(reshape_dims) != TrtDimsNumElements(input_dims)) { in ConvertReshape()2151 TrtDimsNumElements(reshape_dims) != TrtDimsNumElements(input_dims)) { in ConvertReshape()
1219 const int64_t num_elements = TrtDimsNumElements(trt_dims); in AddTestWeights()1689 const int num_input = TrtDimsNumElements(GetTestDims(dims_array)); in TestConvertBiasAdd()1897 const size_t num_inputs = TrtDimsNumElements(GetTestDims(input_dims)); in TestBinaryTensorOpWeightFallback()1898 const size_t num_weights = TrtDimsNumElements(GetTestDims(weights_dims)); in TestBinaryTensorOpWeightFallback()1931 TrtDimsNumElements(GetTestDims(expected_output_dims)); in TestBinaryTensorOpWeightFallback()