Searched refs:TrtTensorDimsNumElements (Results 1 – 3 of 3) sorted by relevance
167 int64_t TrtTensorDimsNumElements(const nvinfer1::Dims& dims);
603 int64_t TrtTensorDimsNumElements(const nvinfer1::Dims& dims) { in TrtTensorDimsNumElements() function611 return TrtTensorDimsNumElements(lhs) == TrtTensorDimsNumElements(rhs); in DimsHaveSameSize()2517 int64_t tensor_size = TrtTensorDimsNumElements(input_tensor->getDimensions()); in ConvertTranspose()
2408 std::vector<float> input_vec(TrtTensorDimsNumElements(actual_output_dims) * in TEST_F()2736 const int num_input = TrtTensorDimsNumElements(GetTestDims(dims_array)); in TEST_P()